Update app.py

app.py

@@ -19,6 +19,8 @@ if torch.cuda.is_available():
 
     # pipe.vae = AutoencoderKL.from_pretrained("IDKiro/sdxs-512-0.9/vae_large")     # use original VAE
     pipe.to("cuda")
+else:
+    pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
 
 # Function 
 @spaces.GPU(enable_queue=True)