Update app.py

app.py

@@ -33,7 +33,7 @@ step = 2
 #ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 #base = "emilianJR/epiCRealism"
 base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
-vae = AutoencoderKL.from_single_file("https://huggingface.co/stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors").to(device, dtype=dtype)
+vae = AutoencoderKL.from_single_file("https://huggingface.co/emilianJR/epiCRealism/vae/diffusion_pytorch_model.safetensors").to(device, dtype=dtype)
 #unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
 adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
 
@@ -142,6 +142,18 @@ def Piper(image,positive,negative,motion):
 
     pipe.to(device)
 
+    if image == None:
+        return pipe(
+            positive,
+            negative_prompt=negative,
+            height=1024,
+            width=576,
+            ip_adapter_image=image.convert("RGB").resize((576,1024)),
+            num_inference_steps=step,
+            guidance_scale=1,
+            num_frames=(fps*time)
+        )
+
     return pipe(
         positive,
         negative_prompt=negative,