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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def load_models():
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!")
 
73
  try:
74
  st.info("Loading Gemma model...")
75
  gemma_pipe = pipeline("text-generation",
76
+ model="google/gemma-2-2b-it",
77
  device=0,
78
  torch_dtype=torch.bfloat16)
79
  st.success("Gemma model loaded successfully!")