Update app.py

app.py

@@ -15,7 +15,7 @@ import spaces
 model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
 vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
-flow_shift = 5.0  # 5.0 for 720P, 3.0 for 480P
+flow_shift = 1.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
 
 
@@ -38,7 +38,7 @@ def generate(prompt, negative_prompt, width=1024, height=1024, num_inference_ste
             width=width,
             num_frames=1,
             num_inference_steps=num_inference_steps,
-            guidance_scale=5.0,
+            guidance_scale=1.0, #5.0
         )
         image = output.frames[0][0]
         image = (image * 255).astype(np.uint8)