Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import cv2
 from PIL import Image, ImageDraw, ImageFont
@@ -15,6 +16,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16").to(device)
 
+@spaces.GPU(duration=200)
 def process_video(video_path, target, progress=gr.Progress()):
     if video_path is None:
         return None, None, "Error: No video uploaded"
@@ -129,7 +131,6 @@ def load_sample_frame(video_path):
     frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     return frame_rgb
 
-@spaces.GPU(duration=200)
 def gradio_app():
     with gr.Blocks() as app:
         gr.Markdown("# Video Object Detection with Owlv2")