Hugging Face's logo

Spaces:
Solo448
/
OCR_MULTILINGUAL-GOT
Running

App Files Community
Solo448 commited on
Commit
7280c03
·
verified ·
1 Parent(s): 84534f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def image_to_base64(image):
34
 
35
  q_model_name = "Qwen/Qwen2-VL-2B-Instruct"
36
  #q_model = Qwen2VLForConditionalGeneration.from_pretrained(q_model_name, torch_dtype="auto").cuda().eval()
37
- q_model = Qwen2VLForConditionalGeneration.from_pretrained(q_model_name, torch_dtype="cpu").eval()
38
  q_processor = AutoProcessor.from_pretrained(q_model_name, trust_remote_code=True)
39
 
40
  def get_qwen_op(image_file, model, processor):
 
34
 
35
  q_model_name = "Qwen/Qwen2-VL-2B-Instruct"
36
  #q_model = Qwen2VLForConditionalGeneration.from_pretrained(q_model_name, torch_dtype="auto").cuda().eval()
37
+ q_model = Qwen2VLForConditionalGeneration.from_pretrained(q_model_name, torch_dtype="auto").eval()
38
  q_processor = AutoProcessor.from_pretrained(q_model_name, trust_remote_code=True)
39
 
40
  def get_qwen_op(image_file, model, processor):