added spaces GPU

agents.py

@@ -110,6 +110,7 @@ def describe_with_voice(state: State) -> State:
     description = processor.decode(generation, skip_special_tokens=True)
 
     state["description"] = description
+    print(description)
 
     return state
 
@@ -153,5 +154,6 @@ def caption_image(state: State) -> State:
     caption = processor.decode(generation, skip_special_tokens=True)
 
     state["caption"] = caption
+    print(caption)
 
     return state

app.py

@@ -1,18 +1,19 @@
 import gradio as gr
 
 from agents import build_graph
+import spaces
 
 # Initialize the graph
 graph = build_graph()
 
 
+@spaces.GPU(duration=60)
 def process_and_display(image, voice):
     # Initialize state
     state = {"image": image, "voice": voice, "caption": "", "description": ""}
 
     # Run the graph
     result = graph.invoke(state)
-    print(result)
 
     # Return the caption and description
     return result["caption"], result["description"]