Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ def initialize_model():
|
|
| 7 |
"""Initialize a small and fast model for CPU"""
|
| 8 |
# Using a tiny model optimized for CPU
|
| 9 |
model_id = "facebook/opt-125m" # Much smaller model (125M parameters)
|
|
|
|
| 10 |
|
| 11 |
try:
|
| 12 |
# Initialize the pipeline directly - more efficient than loading model separately
|
|
|
|
| 7 |
"""Initialize a small and fast model for CPU"""
|
| 8 |
# Using a tiny model optimized for CPU
|
| 9 |
model_id = "facebook/opt-125m" # Much smaller model (125M parameters)
|
| 10 |
+
model_id ="GEB-AGI/geb-1.3b"
|
| 11 |
|
| 12 |
try:
|
| 13 |
# Initialize the pipeline directly - more efficient than loading model separately
|