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