prithivMLmods commited on
Commit
95bf8be
·
verified ·
1 Parent(s): 55a7e0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ dtype = torch.float16 if device.type == "cuda" else torch.float32
92
 
93
  # Gemma3 Model (default for text, image, & video inference)
94
 
95
- gemma3_model_id = "google/gemma-3-4b-it" # alternative: google/gemma-3-12b-it
96
  gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
97
  gemma3_model_id, device_map="auto"
98
  ).eval()
 
92
 
93
  # Gemma3 Model (default for text, image, & video inference)
94
 
95
+ gemma3_model_id = "google/gemma-3-4b-it"
96
  gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
97
  gemma3_model_id, device_map="auto"
98
  ).eval()