Update app.py

app.py

@@ -55,7 +55,7 @@ SCHEDULER_MAP = {
 }
 
 pipe = WanImageToVideoPipeline.from_pretrained(
-    "TestOrganizationPleaseIgnore/WAMU_v2_WAN2.2_I2V_LIGHTNING_preview4",
+    "TestOrganizationPleaseIgnore/WAMU_v2_WAN2.2_I2V_LIGHTNING_preview3",
     torch_dtype=torch.bfloat16,
 ).to('cuda')
 original_scheduler = copy.deepcopy(pipe.scheduler)
@@ -157,7 +157,7 @@ def get_inference_duration(
     width, height = resized_image.size
     factor = num_frames * width * height / BASE_FRAMES_HEIGHT_WIDTH
     step_duration = BASE_STEP_DURATION * factor ** 1.5
-    return 5 + int(steps) * step_duration
+    return 10 + int(steps) * step_duration
 
 
 @spaces.GPU(duration=get_inference_duration)