eong commited on
Commit
6de1a16
·
verified ·
1 Parent(s): 914998b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -79,13 +79,13 @@ final_answer = FinalAnswerTool()
79
  # custom_role_conversions=None,
80
  # )
81
 
82
- #model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="gemini_api"))
83
 
84
- model = LiteLLMModel(
85
- model_id="ollama_chat/qwen2:7b", # Or try other Ollama-supported models
86
- api_base=os.getenv(key='ollama_server_ip'), # Default Ollama local server
87
- num_ctx=8192,
88
- )
89
 
90
  # Import tool from Hub
91
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
79
  # custom_role_conversions=None,
80
  # )
81
 
82
+ model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="gemini_api"))
83
 
84
+ # model = LiteLLMModel(
85
+ # model_id="ollama_chat/qwen2:7b", # Or try other Ollama-supported models
86
+ # api_base=os.getenv(key='ollama_server_ip'), # Default Ollama local server
87
+ # num_ctx=8192,
88
+ # )
89
 
90
  # Import tool from Hub
91
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)