AMfeta99 commited on
Commit
7a56df2
·
verified ·
1 Parent(s): d6307b2

Update app.py

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