frankai98 commited on
Commit
93eb40c
·
verified ·
1 Parent(s): a2ec039

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -72,9 +72,8 @@ def load_models():
72
 
73
  try:
74
  st.info("Loading Gemma model...")
75
- # Try with correct model name - either gemma-3-1b-it or gemma-7b-it
76
  gemma_pipe = pipeline("text-generation",
77
- model="google/gemma-3-1b-it", # Check model name
78
  device=0,
79
  torch_dtype=torch.bfloat16)
80
  st.success("Gemma model loaded successfully!")
 
72
 
73
  try:
74
  st.info("Loading Gemma model...")
 
75
  gemma_pipe = pipeline("text-generation",
76
+ model="google/gemma-3-1b-pt",
77
  device=0,
78
  torch_dtype=torch.bfloat16)
79
  st.success("Gemma model loaded successfully!")