KingNish commited on
Commit
5de9904
1 Parent(s): 29c0cfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
  from threading import Thread
11
 
12
  # Model and Processor Loading (Done once at startup)
13
- MODEL_ID = "Qwen/Qwen2-VL-2B-Instruct"
14
  model = Qwen2VLForConditionalGeneration.from_pretrained(
15
  MODEL_ID,
16
  trust_remote_code=True,
@@ -18,7 +18,7 @@ model = Qwen2VLForConditionalGeneration.from_pretrained(
18
  ).to("cuda").eval()
19
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
20
 
21
- DESCRIPTION = "[Qwen2-VL-2B Demo](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct)"
22
 
23
  image_extensions = Image.registered_extensions()
24
  video_extensions = ("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")
 
10
  from threading import Thread
11
 
12
  # Model and Processor Loading (Done once at startup)
13
+ MODEL_ID = "Qwen/Qwen2-VL-7B-Instruct"
14
  model = Qwen2VLForConditionalGeneration.from_pretrained(
15
  MODEL_ID,
16
  trust_remote_code=True,
 
18
  ).to("cuda").eval()
19
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
20
 
21
+ DESCRIPTION = "[Qwen2-VL-7B Demo](https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct)"
22
 
23
  image_extensions = Image.registered_extensions()
24
  video_extensions = ("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")