Update app.py

app.py

@@ -10,6 +10,7 @@ import torch
 import os
 import gradio as gr
 import numpy as np
+import space
 
 np.float = float  
 np.int = int 
@@ -44,7 +45,7 @@ filename = 'demo.webm'
 
 description = "Yolo model to detect if a person is falling or fallen with deepsort to track how long the subject has fallen.\
             If the duration crosses a threshold of 5s, the bounding box will turn red and the subject be labelled as IMMOBILE."
-
+@spaces.GPU
 def vid_predict(media):
 
     pipeline = DetectionPipeline(