Update app.py

app.py

@@ -419,7 +419,7 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__" or os.getenv("SYSTEM") == "spaces":
     try:
-        get_local_model_gpu()  # triggers GPU allocation during startup
+        get_local_model_gpu(model_id)  # 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)