Update app.py

app.py

@@ -12,7 +12,6 @@ import numpy as np
 from PIL import Image
 import spaces
 
-
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
@@ -40,8 +39,7 @@ def annotate_image(
     output_image = LABEL_ANNOTATOR.annotate(output_image, detections, labels=labels)
     return output_image
 
-
-
+@spaces.GPU
 def process_video(
     input_video,
     labels,
@@ -63,7 +61,6 @@ def process_video(
             # list of dict of {"box": box, "mask":mask, "score":score, "label":label}
             results = query(frame, labels)
 
-            #detections = sv.Detections.empty()
             detections = sv.Detections.from_transformers(results[0])
             final_labels = []
             for id in results[0]["labels"]:
@@ -76,7 +73,6 @@ def process_video(
             sink.write_frame(frame)
     return result_file_path
 
-@spaces.GPU
 def query(image, texts):
   inputs = processor(text=texts, images=image, return_tensors="pt").to(device)
   with torch.no_grad():