Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"
|
| 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
|