Toumaima commited on
Commit
272ec1b
·
verified ·
1 Parent(s): 8527a78

Update app.py

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