openfree commited on
Commit
3d1fae7
·
verified ·
1 Parent(s): 8378e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from loguru import logger
14
  from PIL import Image
15
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
16
 
17
- model_id = os.getenv("MODEL_ID", "google/gemma-3-12b-it")
18
  processor = AutoProcessor.from_pretrained(model_id, padding_side="left")
19
  model = Gemma3ForConditionalGeneration.from_pretrained(
20
  model_id, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager"
 
14
  from PIL import Image
15
  from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
16
 
17
+ model_id = os.getenv("MODEL_ID", "google/gemma-3-27b-it")
18
  processor = AutoProcessor.from_pretrained(model_id, padding_side="left")
19
  model = Gemma3ForConditionalGeneration.from_pretrained(
20
  model_id, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager"