Spaces:
Running
on
Zero
Running
on
Zero
wjm55
commited on
Commit
·
d312ec3
1
Parent(s):
f07ac74
Update to Qwen2.5-VL model and clean up requirements
Browse files- app.py +4 -4
- requirements.txt +1 -3
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import spaces
|
3 |
-
from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
|
4 |
from qwen_vl_utils import process_vision_info
|
5 |
import torch
|
6 |
from PIL import Image
|
@@ -49,15 +49,15 @@ def array_to_image_path(image_array):
|
|
49 |
return full_path
|
50 |
|
51 |
models = {
|
52 |
-
"
|
53 |
|
54 |
}
|
55 |
|
56 |
processors = {
|
57 |
-
"
|
58 |
}
|
59 |
|
60 |
-
DESCRIPTION = "This demo uses[Qwen2-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct)"
|
61 |
|
62 |
kwargs = {}
|
63 |
kwargs['torch_dtype'] = torch.bfloat16
|
|
|
1 |
import gradio as gr
|
2 |
import spaces
|
3 |
+
from transformers import Qwen2VLForConditionalGeneration, AutoProcessor, Qwen2_5_VLForConditionalGeneration
|
4 |
from qwen_vl_utils import process_vision_info
|
5 |
import torch
|
6 |
from PIL import Image
|
|
|
49 |
return full_path
|
50 |
|
51 |
models = {
|
52 |
+
"Qwen2.5-VL-7B-Instruct": Qwen2_5_VLForConditionalGeneration.from_pretrained("Qwen2.5-VL-7B-Instruct", trust_remote_code=True, torch_dtype="auto").cuda().eval()
|
53 |
|
54 |
}
|
55 |
|
56 |
processors = {
|
57 |
+
"Qwen2.5-VL-7B-Instruct": AutoProcessor.from_pretrained("Qwen2.5-VL-7B-Instruct", trust_remote_code=True)
|
58 |
}
|
59 |
|
60 |
+
DESCRIPTION = "This demo uses[Qwen2.5-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-VL-7B-Instruct)"
|
61 |
|
62 |
kwargs = {}
|
63 |
kwargs['torch_dtype'] = torch.bfloat16
|
requirements.txt
CHANGED
@@ -3,8 +3,6 @@ Pillow==10.3.0
|
|
3 |
Requests==2.31.0
|
4 |
torch
|
5 |
torchvision
|
6 |
-
git+https://github.com/huggingface/transformers
|
7 |
accelerate
|
8 |
qwen-vl-utils
|
9 |
-
git+https://github.com/huggingface/transformers.git
|
10 |
-
gliner
|
|
|
3 |
Requests==2.31.0
|
4 |
torch
|
5 |
torchvision
|
6 |
+
git+https://github.com/huggingface/transformers
|
7 |
accelerate
|
8 |
qwen-vl-utils
|
|
|
|