add zeroGPU decoration

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import numpy as np 
 import os 
 import tempfile
+import spaces
 import gradio as gr
 
 import cv2
@@ -61,6 +62,7 @@ def visualize(pred_mask, image_path, work_dir):
     cv2.imwrite(output_path, visual_result)
     return output_path
 
+@spaces.GPU
 def image_vision(image_input_path, prompt):
     image_path = image_input_path
     text_prompts = f"<image>{prompt}"
@@ -88,6 +90,7 @@ def image_vision(image_input_path, prompt):
     else:
         return answer, None
 
+@spaces.GPU(duration=80)
 def video_vision(video_input_path, prompt, video_interval):
     # Open the original video
     cap = cv2.VideoCapture(video_input_path)