prithivMLmods commited on
Commit
16f62b6
·
verified ·
1 Parent(s): 231ec6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -12,10 +12,8 @@ from transformers import (
12
  AutoProcessor,
13
  TextIteratorStreamer,
14
  )
15
- from transformers import Qwen2_5_VLForConditionalGeneration
16
-
17
  # Helper Functions
18
- def progress_bar_html(label: str, primary_color: str = "#4B0082", secondary_color: str = "#9370DB") -> str:
19
  """
20
  Returns an HTML snippet for a thin animated progress bar with a label.
21
  Colors can be customized; default colors are used for Qwen2VL/Aya-Vision.
@@ -60,9 +58,9 @@ def downsample_video(video_path):
60
  return frames
61
 
62
  # Model and Processor Setup
63
- QV_MODEL_ID = "Qwen/Qwen2.5-VL-32B-Instruct"
64
  qwen_processor = AutoProcessor.from_pretrained(QV_MODEL_ID, trust_remote_code=True)
65
- qwen_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
66
  QV_MODEL_ID,
67
  trust_remote_code=True,
68
  torch_dtype=torch.float16
 
12
  AutoProcessor,
13
  TextIteratorStreamer,
14
  )
 
 
15
  # Helper Functions
16
+ def progress_bar_html(label: str, primary_color: str = "#FF4500", secondary_color: str = "#FFA07A") -> str:
17
  """
18
  Returns an HTML snippet for a thin animated progress bar with a label.
19
  Colors can be customized; default colors are used for Qwen2VL/Aya-Vision.
 
58
  return frames
59
 
60
  # Model and Processor Setup
61
+ QV_MODEL_ID = "Qwen/Qwen2-VL-7B-Instruct"
62
  qwen_processor = AutoProcessor.from_pretrained(QV_MODEL_ID, trust_remote_code=True)
63
+ qwen_model = Qwen2VLForConditionalGeneration.from_pretrained(
64
  QV_MODEL_ID,
65
  trust_remote_code=True,
66
  torch_dtype=torch.float16