prithivMLmods commited on
Commit
5373e26
·
verified ·
1 Parent(s): 2aaebb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,16 +8,16 @@ import spaces
8
  import time
9
  from PIL import Image
10
  from threading import Thread
11
- from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, TextIteratorStreamer
12
  from transformers.image_utils import load_image
13
 
14
  #####################################
15
  # 1. Load Qwen2.5-VL Model & Processor
16
  #####################################
17
- MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct" # or "Qwen/Qwen2.5-VL-3B-Instruct"
18
 
19
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
20
- model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
21
  MODEL_ID,
22
  trust_remote_code=True,
23
  torch_dtype=torch.bfloat16
 
8
  import time
9
  from PIL import Image
10
  from threading import Thread
11
+ from transformers import AutoProcessor, Gemma3ForConditionalGeneration,, TextIteratorStreamer
12
  from transformers.image_utils import load_image
13
 
14
  #####################################
15
  # 1. Load Qwen2.5-VL Model & Processor
16
  #####################################
17
+ MODEL_ID = "google/gemma-3-12b-it" # or "Qwen/Qwen2.5-VL-3B-Instruct"
18
 
19
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
20
+ model = Gemma3ForConditionalGeneration.from_pretrained(
21
  MODEL_ID,
22
  trust_remote_code=True,
23
  torch_dtype=torch.bfloat16