Update app.py

app.py

@@ -20,6 +20,7 @@ import tempfile
 import shutil
 import subprocess
 import fractions
+import spaces
 
 # Suppress TensorFlow warnings
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
@@ -349,6 +350,8 @@ def get_random_face_sample(organized_faces_folder, largest_cluster, output_folde
         return output_path
     return None
 
+@spaces.GPU(duration=300)  # Set to 5 minutes (300 seconds)
+
 def process_video(video_path, num_anomalies, num_components, desired_fps, batch_size, progress=gr.Progress()):
     output_folder = "output"
     os.makedirs(output_folder, exist_ok=True)