Spaces:
Running
Running
Commit
·
4667ca1
1
Parent(s):
226eb77
3B model
Browse files- app.py +1 -1
- requirements.txt +3 -3
app.py
CHANGED
@@ -14,7 +14,7 @@ from pytube import YouTube # new import for handling YouTube links
|
|
14 |
# ----------------------------------------
|
15 |
# 1. Initialize the Qwen 2.5 VL Model (7B) for CPU-only
|
16 |
# ----------------------------------------
|
17 |
-
model_path = "Qwen/Qwen2.5-VL-
|
18 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
19 |
model_path,
|
20 |
torch_dtype=torch.float16 # use float32 for CPU
|
|
|
14 |
# ----------------------------------------
|
15 |
# 1. Initialize the Qwen 2.5 VL Model (7B) for CPU-only
|
16 |
# ----------------------------------------
|
17 |
+
model_path = "Qwen/Qwen2.5-VL-3B-Instruct"
|
18 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
19 |
model_path,
|
20 |
torch_dtype=torch.float16 # use float32 for CPU
|
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
# Core dependencies from Qwen 2.5 VL
|
2 |
-
gradio
|
3 |
-
gradio_client
|
4 |
qwen-vl-utils
|
5 |
transformers-stream-generator==0.0.4
|
6 |
torch==2.4.0
|
@@ -13,7 +13,7 @@ av
|
|
13 |
# flash-attn==2.6.1
|
14 |
|
15 |
# Additional dependencies for video processing and utilities
|
16 |
-
decord
|
17 |
numpy
|
18 |
Pillow
|
19 |
requests
|
|
|
1 |
# Core dependencies from Qwen 2.5 VL
|
2 |
+
gradio
|
3 |
+
gradio_client
|
4 |
qwen-vl-utils
|
5 |
transformers-stream-generator==0.0.4
|
6 |
torch==2.4.0
|
|
|
13 |
# flash-attn==2.6.1
|
14 |
|
15 |
# Additional dependencies for video processing and utilities
|
16 |
+
decord #use decord for linux or other OS
|
17 |
numpy
|
18 |
Pillow
|
19 |
requests
|