Update app.py

app.py

@@ -34,7 +34,8 @@ os.system(f"wget {' '.join(embeddings)}")
 os.chdir("../models/Stable-diffusion")
 os.system(f"wget {' '.join(models)}")
 files = os.listdir()
-default_vae = [f for f in files if f.endswith(".vae.pt")][0]
+default_vae = [f for f in files if f.endswith(".vae.pt")]
+default_vae = default_vae[0] if default_vae else "None"
 default_model = [f for f in files if f.endswith(".safetensors") or f.endswith(".ckpt")][0]
 os.chdir("../..")