AMfeta99 commited on
Commit
d6307b2
·
verified ·
1 Parent(s): 22ecf74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
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
  # =========================================================