Facelook commited on
Commit
321fb3b
·
1 Parent(s): 99fefaa

Switched to faster model.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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)