oddadmix commited on
Commit
618ee0a
·
verified ·
1 Parent(s): 7baa28c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,8 +15,9 @@ def load_model():
15
  print("Loading model...")
16
  model_id = "google/gemma-3n-e4b-it"
17
 
 
18
  model = Gemma3nForConditionalGeneration.from_pretrained(
19
- model_id,
20
  device_map="auto",
21
  torch_dtype=torch.bfloat16,
22
  ).eval()
 
15
  print("Loading model...")
16
  model_id = "google/gemma-3n-e4b-it"
17
 
18
+
19
  model = Gemma3nForConditionalGeneration.from_pretrained(
20
+ "oddadmix/gemma-4b-egyptian-code-switching-b4-g2-merged",
21
  device_map="auto",
22
  torch_dtype=torch.bfloat16,
23
  ).eval()