Update app.py

app.py

@@ -24,6 +24,7 @@ import time
 import timeout_decorator
 import socket
 import ipaddress
+from huggingface_hub import hf_hub_download
 from huggingface_hub import spaces
 
 # Setup logging
@@ -59,6 +60,19 @@ MAX_IMAGE_RES = (1920, 1080)
 MAX_VIDEO_MB = 50
 MAX_VIDEO_DURATION = 30  # seconds
 
+@spaces.GPU
+def preload_models():
+    """
+    This function is needed to activate ZeroGPU. It must be decorated with @spaces.GPU.
+    It can be used to warm up models or load them into memory.
+    """
+    from registry import get_model
+    print("Warming up models for ZeroGPU...")
+    get_model("detection", "yolov5n-seg", device="cpu")
+    get_model("segmentation", "deeplabv3_resnet50", device="cpu")
+    get_model("depth", "midas_v21_small_256", device="cpu")
+
+
 # Utility Functions
 def format_error(message):
     """Formats error messages for consistent user feedback."""
@@ -461,8 +475,4 @@ with gr.Blocks() as demo:
     )
 
 # Launch the Gradio App
-@spaces.GPU
-def launch_app():
-    demo.launch()
-
-launch_app()
+demo.launch()