Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -189,11 +189,11 @@ def resize_image_for_display(img_array, max_width=500):
|
|
189 |
return np.array(img)
|
190 |
|
191 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
192 |
-
"weihongliang/RC-Qwen2VL-
|
193 |
torch_dtype=torch.float16,
|
194 |
device_map="auto"
|
195 |
)
|
196 |
-
processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-
|
197 |
|
198 |
# Use Qwen model for Q&A
|
199 |
def qwen_qa(image_path, question, names, bounding_boxes, en_wiki_pedia_links, en_intros):
|
|
|
189 |
return np.array(img)
|
190 |
|
191 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
192 |
+
"weihongliang/RC-Qwen2VL-7b",
|
193 |
torch_dtype=torch.float16,
|
194 |
device_map="auto"
|
195 |
)
|
196 |
+
processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
|
197 |
|
198 |
# Use Qwen model for Q&A
|
199 |
def qwen_qa(image_path, question, names, bounding_boxes, en_wiki_pedia_links, en_intros):
|