Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -157,7 +157,7 @@ div.no-padding { padding: 0 !important; }
|
|
| 157 |
"""
|
| 158 |
|
| 159 |
MAX_MAX_NEW_TOKENS = 4096
|
| 160 |
-
DEFAULT_MAX_NEW_TOKENS =
|
| 161 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 162 |
|
| 163 |
print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
|
|
@@ -373,7 +373,7 @@ def generate_gif(text: str, gif_path: str, max_new_tokens: int = 1024, temperatu
|
|
| 373 |
time.sleep(0.01)
|
| 374 |
yield buffer, buffer
|
| 375 |
|
| 376 |
-
image_examples = [["Perform OCR on the image precisely
|
| 377 |
["Caption the image. Describe the safety measures shown in the image. Conclude whether the situation is (safe or unsafe)...", "examples/images/2.jpg"],
|
| 378 |
["Solve the problem...", "examples/images/3.png"]]
|
| 379 |
video_examples = [["Explain the Ad video in detail.", "examples/videos/1.mp4"],
|
|
|
|
| 157 |
"""
|
| 158 |
|
| 159 |
MAX_MAX_NEW_TOKENS = 4096
|
| 160 |
+
DEFAULT_MAX_NEW_TOKENS = 1024
|
| 161 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 162 |
|
| 163 |
print("CUDA_VISIBLE_DEVICES=", os.environ.get("CUDA_VISIBLE_DEVICES"))
|
|
|
|
| 373 |
time.sleep(0.01)
|
| 374 |
yield buffer, buffer
|
| 375 |
|
| 376 |
+
image_examples = [["Perform OCR on the image precisely...", "examples/images/1.jpg"],
|
| 377 |
["Caption the image. Describe the safety measures shown in the image. Conclude whether the situation is (safe or unsafe)...", "examples/images/2.jpg"],
|
| 378 |
["Solve the problem...", "examples/images/3.png"]]
|
| 379 |
video_examples = [["Explain the Ad video in detail.", "examples/videos/1.mp4"],
|