nico-s commited on
Commit
869fe74
·
verified ·
1 Parent(s): 9017746

added mistral

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -36,17 +36,17 @@ def get_current_time_in_timezone(timezone: str) -> str:
36
 
37
 
38
  final_answer = FinalAnswerTool()
39
- #model = HfApiModel(
40
- # max_tokens=2096,
41
- # temperature=0.5,
42
- # model_id='meta-llama/Llama-3.1-8B-Instruct',# it is possible that this model may be overloaded
43
- # custom_role_conversions=None,
44
- #)
45
- model = OpenAIServerModel(
46
- model_id="gpt-4o-mini",
47
- #api_base="https://api.together.xyz/v1/", # Leave this blank to query OpenAI servers.
48
- api_key=os.environ["OPENAI_API_KEY"], # Switch to the API key for the server you're targeting.
49
  )
 
 
 
 
 
50
 
51
  # Import tool from Hub
52
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
36
 
37
 
38
  final_answer = FinalAnswerTool()
39
+ model = HfApiModel(
40
+ max_tokens=2096,
41
+ temperature=0.5,
42
+ model_id='mistralai/Mistral-7B-v0.1',# it is possible that this model may be overloaded
43
+ custom_role_conversions=None,
 
 
 
 
 
44
  )
45
+ #model = OpenAIServerModel(
46
+ # model_id="gpt-4o-mini",
47
+ #api_base="https://api.together.xyz/v1/", # Leave this blank to query OpenAI servers.
48
+ # api_key=os.environ["OPENAI_API_KEY"], # Switch to the API key for the server you're targeting.
49
+ #)
50
 
51
  # Import tool from Hub
52
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)