prithivMLmods commited on
Commit
277fe06
·
verified ·
1 Parent(s): f9ecde0

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 2048
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 and reconstruct it correctly...", "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"],
 
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"],