Update app.py
Browse files
app.py
CHANGED
@@ -129,7 +129,7 @@ image_generation_tool = Tool.from_space(
|
|
129 |
#image_generation_tool= TextToImageTool()
|
130 |
#image_generation_tool = WrappedTextToImageTool()
|
131 |
search_tool = DuckDuckGoSearchTool()
|
132 |
-
print('iiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiii')
|
133 |
#llm_engine = InferenceClientModel("Qwen/Qwen2.5-72B-Instruct")
|
134 |
#llm_engine = TransformersModel(
|
135 |
# model_id="Qwen/Qwen2.5-72B-Instruct",
|
@@ -137,11 +137,10 @@ print('iiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiii
|
|
137 |
# max_new_tokens=5000,
|
138 |
#)
|
139 |
|
140 |
-
from smolagents import LiteLLMModel
|
141 |
-
|
142 |
-
llm_engine = LiteLLMModel(model_id="Qwen/Qwen2.5-72B-Instruct", temperature=0.2, max_tokens=5000)
|
143 |
-
print('aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa')
|
144 |
|
|
|
|
|
145 |
agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
|
146 |
|
147 |
# =========================================================
|
|
|
129 |
#image_generation_tool= TextToImageTool()
|
130 |
#image_generation_tool = WrappedTextToImageTool()
|
131 |
search_tool = DuckDuckGoSearchTool()
|
132 |
+
#print('iiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiii')
|
133 |
#llm_engine = InferenceClientModel("Qwen/Qwen2.5-72B-Instruct")
|
134 |
#llm_engine = TransformersModel(
|
135 |
# model_id="Qwen/Qwen2.5-72B-Instruct",
|
|
|
137 |
# max_new_tokens=5000,
|
138 |
#)
|
139 |
|
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 |
# =========================================================
|