Update app.py
Browse files
app.py
CHANGED
@@ -76,7 +76,6 @@ class WrappedTextToImageTool(Tool):
|
|
76 |
return self.client.text_to_image(prompt)
|
77 |
|
78 |
'''
|
79 |
-
from huggingface_hub import InferenceClient
|
80 |
|
81 |
'''
|
82 |
class TextToImageTool(Tool):
|
@@ -140,7 +139,9 @@ search_tool = DuckDuckGoSearchTool()
|
|
140 |
#from smolagents import LiteLLMModel
|
141 |
|
142 |
#llm_engine = LiteLLMModel(model_id="Qwen/Qwen2.5-72B-Instruct", temperature=0.2, max_tokens=5000)
|
143 |
-
llm_engine=InferenceClientModel()
|
|
|
|
|
144 |
agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
|
145 |
|
146 |
# =========================================================
|
|
|
76 |
return self.client.text_to_image(prompt)
|
77 |
|
78 |
'''
|
|
|
79 |
|
80 |
'''
|
81 |
class TextToImageTool(Tool):
|
|
|
139 |
#from smolagents import LiteLLMModel
|
140 |
|
141 |
#llm_engine = LiteLLMModel(model_id="Qwen/Qwen2.5-72B-Instruct", temperature=0.2, max_tokens=5000)
|
142 |
+
#llm_engine=InferenceClientModel()
|
143 |
+
|
144 |
+
LL_engine = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
|
145 |
agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
|
146 |
|
147 |
# =========================================================
|