KingNish commited on
Commit
15a7c22
Β·
verified Β·
1 Parent(s): 84ca056

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +1 -1
chatbot.py CHANGED
@@ -26,7 +26,7 @@ from gradio_client import Client, file
26
  from groq import Groq
27
 
28
  # Model and Processor Loading (Done once at startup)
29
- MODEL_ID = "Qwen/Qwen2-VL-7B-Instruct"
30
  model = Qwen2VLForConditionalGeneration.from_pretrained(MODEL_ID, trust_remote_code=True, torch_dtype=torch.float16).to("cuda").eval()
31
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
32
 
 
26
  from groq import Groq
27
 
28
  # Model and Processor Loading (Done once at startup)
29
+ MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct"
30
  model = Qwen2VLForConditionalGeneration.from_pretrained(MODEL_ID, trust_remote_code=True, torch_dtype=torch.float16).to("cuda").eval()
31
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
32