initial test

app.py

@@ -9,6 +9,7 @@ import torch
 from pathlib import Path
 import time
 import torch
+import spaces
 
 
 from video_highlight_detector import (
@@ -37,6 +38,7 @@ def add_watermark(video_path: str, output_path: str):
         -codec:a copy {output_path}"""
     os.system(command)
 
+@spaces.GPU
 def process_video(
     video_path: str,
     progress = gr.Progress()
@@ -162,10 +164,8 @@ def create_ui(examples_path: str):
 
 if __name__ == "__main__":
     # Initialize CUDA
-    if not torch.cuda.is_available():
-        raise RuntimeError("This application requires a GPU to run")
-    torch.cuda.init()
-    torch.cuda.empty_cache()
-    
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    zero = torch.Tensor([0]).to(device)
+
     app = create_ui("video_spec.json")
     app.launch()