Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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!")
|