Update app.py

app.py

@@ -8,6 +8,8 @@ from langchain import HuggingFaceHub
 
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 import torch
+from diffusers import DiffusionPipeline
+
 
 
 # Suppressing all warnings
@@ -68,13 +70,14 @@ def txt2speech(text):
 
 # text-to- image
 def txt2img(text, style="realistic"):
-    model_id = "stabilityai/stable-diffusion-2"
+    #model_id = "stabilityai/stable-diffusion-2"
+    pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
 
     # Use the Euler scheduler here instead
-    scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
+    #scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+    #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
     #pipe = pipe.to("cuda")
-    image = pipe(prompt = text, guidance_scale = 7.5).images[0]
+    image = pipeline(prompt = text, guidance_scale = 7.5).images[0]
     return image