Synnove commited on
Commit
a6d15ad
·
verified ·
1 Parent(s): 227876b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import inspect
5
  import yaml
6
  import pandas as pd
7
  #import smolagents #to test
8
- from smolagents import CodeAgent, InferenceClientModel, load_tool, tool #DuckDuckGoSearchTool,
9
  from huggingface_hub import InferenceClient
10
  import json
11
  from final_answer import FinalAnswerTool
@@ -56,7 +56,7 @@ def load_questions_from_file(filepath="questions.json"):
56
  model = InferenceClientModel(
57
  max_tokens=1024, #2096,
58
  temperature=0.5,
59
- model_id= "HuggingFaceH4/zephyr-7b-beta", #'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
60
  custom_role_conversions=None,
61
  )
62
 
 
5
  import yaml
6
  import pandas as pd
7
  #import smolagents #to test
8
+ from smolagents import CodeAgent, InferenceClientModel, load_tool, tool, LiteLLMModel #DuckDuckGoSearchTool,
9
  from huggingface_hub import InferenceClient
10
  import json
11
  from final_answer import FinalAnswerTool
 
56
  model = InferenceClientModel(
57
  max_tokens=1024, #2096,
58
  temperature=0.5,
59
+ model_id= 'mistralai/Mistral-7B-Instruct-v0.2', #'Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded #"HuggingFaceH4/zephyr-7b-beta" weeird
60
  custom_role_conversions=None,
61
  )
62