Spaces:

artificialguybr
/

qwen-vl

Build error

artificialguybr commited on Sep 25, 2023

Commit

3944bd4

1 Parent(s): 0d12e2f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ import re
 import requests
 from io import BytesIO
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-14B-Chat-Int4", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-14B-Chat-Int4", device_map="auto", trust_remote_code=True).eval()
 def generate_predictions(image_input, text_input, with_grounding):
     user_image_path = "/tmp/user_input_test_image.jpg"

 import requests
 from io import BytesIO
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-VL-Chat-Int4", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-VL-Chat-Int4", device_map="auto", trust_remote_code=True).eval()
 def generate_predictions(image_input, text_input, with_grounding):
     user_image_path = "/tmp/user_input_test_image.jpg"