Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import random
 
 import spaces #[uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeXLline #,AutoencoderTiny
+from diffusers import StableDiffusionXLPipeline #,AutoencoderTiny
 import torch
 #from diffusers import AutoencoderTiny, StableDiffusionPipeline , DPMSolverMultistepScheduler ,EulerDiscreteScheduler
 from huggingface_hub import login
@@ -57,8 +57,8 @@ var_1="nota-ai/bk-sdm-base-2m"
 var_2="nota-ai/bk-sdm-small"
 
 
-pipe = DiffusionPipeXLline.from_pretrained(
-     var_2, torch_dtype=torch_dtype, use_safetensors=True)
+pipe = StableDiffusionXLPipeline.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)
 #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)