Update app.py

app.py

@@ -7,7 +7,7 @@ import sys
 import FFV1MT_MS
 import flow_tools
 import transformers
-
+@spaces.GPU
 def process_images(videos, x, y):
     # read video file
     cap = cv2.VideoCapture(videos)
@@ -87,7 +87,7 @@ if __name__ =='__main__':
     # # Tesla T4
     
     model = FFV1MT_MS.FFV1DNN()
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
     
     print('Number fo parameters: {}'.format(model.num_parameters()))
     model.to(device)