HenryY2023 commited on
Commit
6475d13
·
verified ·
1 Parent(s): a667ae8
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -125,12 +125,18 @@ final_answer = FinalAnswerTool()
125
  #model_id='google/gemma-3-1b-it',
126
  #custom_role_conversions=None,
127
  #)
 
 
 
 
 
128
  model=LiteLLMModel(
129
- model_id = "ollama/qwen2:7b",
130
- api_base="http://0.0.0.0:11434",
131
  num_ctx=8192,
132
  )
133
 
 
134
  # Import tool from Hub
135
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
136
 
 
125
  #model_id='google/gemma-3-1b-it',
126
  #custom_role_conversions=None,
127
  #)
128
+ #model=LiteLLMModel(
129
+ # model_id = "ollama/qwen2:7b",
130
+ # api_base="http://0.0.0.0:11434",
131
+ # num_ctx=8192,
132
+ #)
133
  model=LiteLLMModel(
134
+ model_id = "gemini-2.0-flash",
135
+ api_base = "https://generativelanguage.googleapis.com/v1beta/models",
136
  num_ctx=8192,
137
  )
138
 
139
+
140
  # Import tool from Hub
141
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
142