Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from datetime import datetime
 import tempfile
@@ -5,6 +6,9 @@ import os
 import json
 import torch
 import gc
+def debug():
+    torch.randn(10).cuda()
+debug()
 from PIL import Image
 from decord import VideoReader, cpu
 from yolo_detection import (
@@ -29,6 +33,7 @@ global_media_path = None
 tmp_dir = os.path.join('.', 'tmp')
 os.makedirs(tmp_dir, exist_ok=True)
 
+@spaces.GPU
 def process_diary(day, date, total_people, total_machinery, machinery_types, activities, media):
     """Process the site diary entry"""
     global global_activities, global_media_path
@@ -460,4 +465,4 @@ with gr.Blocks(title="Digital Site Diary", css="") as demo:
     """)
 
 if __name__ == "__main__":
-    demo.launch(share=True, allowed_paths=["./tmp"])
+    demo.launch(allowed_paths=["./tmp"])