Update app.py

app.py

@@ -60,17 +60,18 @@ 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")
+
+# @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