Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
20 |
class BasicAgent:
|
21 |
def __init__(self):
|
22 |
print("BasicAgent initialized.")
|
23 |
-
|
|
|
24 |
|
25 |
SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
|
26 |
finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
|
|
|
20 |
class BasicAgent:
|
21 |
def __init__(self):
|
22 |
print("BasicAgent initialized.")
|
23 |
+
api_token = os.getenv("HF_TOKEN")
|
24 |
+
self.model = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1", 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].
|