Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ class BasicAgent:
|
|
65 |
# Initialize the model
|
66 |
print("🤖 Initializing TransformersModel...")
|
67 |
self.model = TransformersModel(
|
68 |
-
model_id="google/gemma-3-
|
69 |
torch_dtype=torch.bfloat16,
|
70 |
device_map="auto",
|
71 |
)
|
|
|
65 |
# Initialize the model
|
66 |
print("🤖 Initializing TransformersModel...")
|
67 |
self.model = TransformersModel(
|
68 |
+
model_id="google/gemma-3-4b-it",
|
69 |
torch_dtype=torch.bfloat16,
|
70 |
device_map="auto",
|
71 |
)
|