Update app.py

app.py

@@ -16,6 +16,7 @@ from cover.datasets import UnifiedFrameSampler, spatial_temporal_view_decomposit
 from cover.models import COVER
 
 import pandas as pd
+import spaces
 
 mean, std = (
     torch.FloatTensor([123.675, 116.28, 103.53]),
@@ -110,6 +111,7 @@ def create_bar_chart(scores, comparisons):
     return image_path
 
 
+@spaces.GPU
 def inference_one_video(input_video):
     """
     BASIC SETTINGS