Summary

app/event_handlers/submit.py

@@ -5,6 +5,7 @@ Description: Event handler for Gradio app to submit.
 License: MIT License
 """
 
+import spaces
 import torch
 import pandas as pd
 import cv2
@@ -40,6 +41,7 @@ from app.load_models import VideoFeatureExtractor
 from app.components import html_message
 
 
+@spaces.GPU
 def event_handler_submit(
     video: str,
 ) -> tuple[gr.HTML, gr.Plot, gr.Plot, gr.Plot, gr.Plot]:

requirements.txt

@@ -1,5 +1,6 @@
 gradio==5.4.0
 gradio_client==1.4.2
+spaces==0.30.4
 polars==1.12.0
 torch==2.2.2
 torchaudio==2.2.2