Update app.py

app.py

@@ -3,7 +3,10 @@ import cv2
 import numpy as np
 from transformers import AutoTokenizer, PaliGemmaForConditionalGeneration, PaliGemmaProcessor
 import torch
+import spaces
 import gradio as gr
+import jax
+import jax.numpy as jnp
 
 # Load PaliGemma
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -25,6 +28,7 @@ def draw_bounding_box(draw, coordinates, label, width, height):
     # Draw bounding box
     draw.rectangle([(x1, y1), (x2, y2)], outline="red", width=2)
 
+@spaces.GPU
 def process_video(video_path, input_text):
     cap = cv2.VideoCapture(video_path)
     fourcc = cv2.VideoWriter_fourcc(*'XVID')