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)
|