Update app.py

app.py

@@ -225,12 +225,11 @@ def main():
 
     repo="stabilityai/sd-vae-ft-mse-original"
     ckpt="vae-ft-mse-840000-ema-pruned.safetensors"
-    vae = model(load_file(hf_hub_download(repo, ckpt), device=device)
-    vae = "./vae"
+    vae = model(load_file(hf_hub_download(repo, ckpt), device=device))
 
     repo="ByteDance/SDXL-Lightning"
     ckpt=f"sdxl_lightning_{step}step_unet.safetensors"
-    unet = model(load_file(hf_hub_download(repo, ckpt), device=device)
+    unet = model(load_file(hf_hub_download(repo, ckpt), device=device))
     
     #repo = "SG161222/Realistic_Vision_V6.0_B1_noVAE"