Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import random
 
 import spaces #[uncomment to use ZeroGPU]
-#from diffusers import DiffusionPipeline ,AutoencoderTiny
+from diffusers import DiffusionPipeline ,AutoencoderTiny
 import torch
 from diffusers import AutoencoderTiny, StableDiffusionPipeline , DPMSolverMultistepScheduler
 from huggingface_hub import login
@@ -12,8 +12,8 @@ a=os.getenv('hf_key')
 login(token=a )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-#model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
-model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+#model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 
 
 
@@ -57,7 +57,7 @@ var_1="nota-ai/bk-sdm-base-2m"
 var_2="nota-ai/bk-sdm-small"
 
 
-pipe = StableDiffusionPipeline.from_pretrained(
+pipe = DiffusionPipeline.from_pretrained(
      model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
 #pipe.vae = AutoencoderTiny.from_pretrained(
 #    "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)