Update app.py

app.py

@@ -418,4 +418,11 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__" or os.getenv("SYSTEM") == "spaces":
+    if __name__ == "__main__" or os.getenv("SYSTEM") == "spaces":
+    try:
+        load_model()  # triggers GPU allocation during startup
+    except Exception as e:
+        # don't crash the app if warmup fails; logs will show details
+        print("Warmup failed:", e)
+
     demo.launch()