Update app.py

app.py

@@ -3,7 +3,7 @@ try:
 except:
     import os
     os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
-# import spaces
+import spaces
 import cv2
 import supervision as sv
 import torch
@@ -51,7 +51,7 @@ def predict_frame(frame,_):
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
     return out.get_image()
 
-# @spaces.GPU
+@spaces.GPU
 def inference(image_url, image, min_score):
   if not torch.cuda.is_available():
     cfg.MODEL.DEVICE = "cpu"
@@ -77,7 +77,7 @@ def inference(image_url, image, min_score):
 
     return out.get_image()
 
-# @spaces.GPU
+@spaces.GPU
 def process_vid(video_path):
 
   cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.3