Switched to faster model.
Browse files
app.py
CHANGED
|
@@ -17,7 +17,8 @@ class BasicAgent:
|
|
| 17 |
def __init__(self):
|
| 18 |
print("BasicAgent initialized.")
|
| 19 |
# Try multiple models in order of preference with better error handling
|
| 20 |
-
self.model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
|
|
|
| 21 |
self.hf_token = os.getenv("HF_TOKEN") # Get token from environment if available
|
| 22 |
|
| 23 |
# List of fallback models (free models that don't require authentication)
|
|
|
|
| 17 |
def __init__(self):
|
| 18 |
print("BasicAgent initialized.")
|
| 19 |
# Try multiple models in order of preference with better error handling
|
| 20 |
+
#self.model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 21 |
+
self.model_name = "https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud"
|
| 22 |
self.hf_token = os.getenv("HF_TOKEN") # Get token from environment if available
|
| 23 |
|
| 24 |
# List of fallback models (free models that don't require authentication)
|