Toumaima commited on
Commit
b85e579
·
verified ·
1 Parent(s): 296ef3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ class BasicAgent:
21
  def __init__(self):
22
  print("BasicAgent initialized.")
23
  api_token = os.getenv("HF_TOKEN")
24
- self.model = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1", api_token=api_token) # Using the correct model name here
25
-
26
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
27
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
28
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated
 
21
  def __init__(self):
22
  print("BasicAgent initialized.")
23
  api_token = os.getenv("HF_TOKEN")
24
+ self.model = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.1") # Using the correct model name here
25
+ self.model.set_api_token(api_token)
26
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
27
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
28
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated