rasao commited on
Commit
75f5e1d
·
verified ·
1 Parent(s): f15c8c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import json
3
  import os
4
- from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool, Tool
5
  from PIL import Image
6
  from transformers import pipeline
7
 
@@ -33,7 +33,7 @@ class VisionTool(Tool):
33
  # Agent Initialization
34
 
35
  hf_token = os.getenv("HF_TOKEN")
36
- model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct",token=hf_token)
37
 
38
  agent = CodeAgent(
39
  tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), TranscriptionTool(), VisionTool()],
 
1
  import gradio as gr
2
  import json
3
  import os
4
+ from smolagents import CodeAgent, HfModel, DuckDuckGoSearchTool, VisitWebpageTool, Tool
5
  from PIL import Image
6
  from transformers import pipeline
7
 
 
33
  # Agent Initialization
34
 
35
  hf_token = os.getenv("HF_TOKEN")
36
+ model = HfModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct",token=hf_token)
37
 
38
  agent = CodeAgent(
39
  tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), TranscriptionTool(), VisionTool()],