cowrycode commited on
Commit
d0ebe0e
·
verified ·
1 Parent(s): 07bd3e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -38,7 +38,12 @@ class BasicAgent:
38
  "<answer>\n"
39
  "Once you output '<answer>', stop reasoning and do not call any tool.\n"
40
  )
41
- self.llm = OpenAI(model="gpt-4o", temperature=0.1, api_key=OpenAI_key)
 
 
 
 
 
42
  # self.system_prompt = (
43
  # "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
44
  # "You MUST use the tools provided to answer the user's question. Do not answer from your own knowledge.\n"
 
38
  "<answer>\n"
39
  "Once you output '<answer>', stop reasoning and do not call any tool.\n"
40
  )
41
+ self.llm = HuggingFaceInferenceAPI(
42
+ model_name="deepseek-ai/DeepSeek-R1-0528",
43
+ token=HF_key,
44
+ provider="auto"
45
+ )
46
+ #self.llm = OpenAI(model="gpt-4o", temperature=0.1, api_key=OpenAI_key)
47
  # self.system_prompt = (
48
  # "You are a helpful AI assistant completing GAIA benchmark tasks.\n"
49
  # "You MUST use the tools provided to answer the user's question. Do not answer from your own knowledge.\n"