Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import cv2
 import matplotlib.pyplot as plt
 import random
+import spaces
 import time
 from PIL import Image
 from threading import Thread
@@ -26,6 +27,7 @@ model.eval()
 #####################################
 # 2. Helper Function: Downsample Video
 #####################################
+
 def downsample_video(video_path, num_frames=10):
     """
     Downsamples the video file to `num_frames` evenly spaced frames.
@@ -54,6 +56,7 @@ def downsample_video(video_path, num_frames=10):
 #####################################
 # 3. The Inference Function
 #####################################
+@space.GPU
 def video_inference(video_file, duration):
     """
     - Takes a recorded video file and a chosen duration (string).