frankai98 commited on
Commit
a2ec039
·
verified ·
1 Parent(s): 549832e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,9 +72,9 @@ def load_models():
72
 
73
  try:
74
  st.info("Loading Gemma model...")
75
- # Try with correct model name - either gemma-2-1b-it or gemma-7b-it
76
  gemma_pipe = pipeline("text-generation",
77
- model="google/gemma-2-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
+ # 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!")