Spaces:
Runtime error
Runtime error
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 |
# =========================================================
|