Update app.py

app.py

@@ -70,14 +70,14 @@ def txt2speech(text):
 
 # text-to- image
 def txt2img(text, style="realistic"):
-    #model_id = "stabilityai/stable-diffusion-2"
-    pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
+    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)
-    #pipe = pipe.to("cuda")
-    image = pipeline(prompt = text, guidance_scale = 7.5).images[0]
+    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]
     return image