Spaces:
Running
Running
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
|