baconnier commited on
Commit
d017a04
·
verified ·
1 Parent(s): 85ed65a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,7 +14,9 @@ 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-4b-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-4b-it")
18
+ model_id = os.getenv("MODEL_ID", "baconnier/Napoleon_4B_V0.0")
19
+
20
  processor = AutoProcessor.from_pretrained(model_id, padding_side="left")
21
  model = Gemma3ForConditionalGeneration.from_pretrained(
22
  model_id, device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="eager"