Update app.py

app.py

@@ -9,4 +9,18 @@ demo = gr.Interface(
     outputs=gr.Textbox(label="Greeting"),
 )
 
+from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
+import torch
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-2-1",
+    torch_dtype=torch.float16
+)
+
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(
+    pipe.scheduler.config
+)
+
+pipe = pipe.to("cuda")
+
 demo.launch()