Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|
12 |
from transformers.image_utils import load_image
|
13 |
|
14 |
#####################################
|
15 |
# 1. Load Qwen2.5-VL Model & Processor
|
16 |
#####################################
|
17 |
-
MODEL_ID = "
|
18 |
|
19 |
processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
|
20 |
-
model =
|
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
|