weihongliang commited on
Commit
833b0c3
·
verified ·
1 Parent(s): 1225555

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-2b",
193
  torch_dtype=torch.float16,
194
  device_map="auto"
195
  )
196
- processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-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):
 
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):