Update app.py

app.py

@@ -136,7 +136,7 @@ def load_models():
         #wan_pipe.to(device)
         #wan_pipe.to(dtype=torch.bfloat16)
 
-load_models()
+
 # =========================
 # Renderer
 # =========================
@@ -268,6 +268,7 @@ def build_estimate_rel(x, y, z, phi, theta):
 @spaces.GPU
 def infer(image, prompt, seed):
 
+    load_models()
 
     img = image.convert("RGB")