akhaliq HF staff commited on
Commit
ca705d7
·
verified ·
1 Parent(s): d17c3c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
2
  from smolagents import CodeAgent, HfApiModel, Tool
3
  from PIL import Image
4
  import os
5
- import spaces
6
 
7
  # Define the image generation tool
8
  image_generation_tool = Tool.from_space(
@@ -17,7 +16,6 @@ model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct") # Replace with your model
17
  # Create the Smol Agent
18
  agent = CodeAgent(tools=[image_generation_tool], model=model)
19
 
20
- @spaces.GPU(duration=120)
21
  # Define the chatbot function
22
  def chat_with_agent(user_input, history):
23
  if history is None:
 
2
  from smolagents import CodeAgent, HfApiModel, Tool
3
  from PIL import Image
4
  import os
 
5
 
6
  # Define the image generation tool
7
  image_generation_tool = Tool.from_space(
 
16
  # Create the Smol Agent
17
  agent = CodeAgent(tools=[image_generation_tool], model=model)
18
 
 
19
  # Define the chatbot function
20
  def chat_with_agent(user_input, history):
21
  if history is None: