Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import torch
-from diffusers import AutoPipelineForText2Image
+from diffusers import StableDiffusionPipeline
 
-model_id = "stabilityai/sd-turbo"
+model_id = "runwayml/stable-diffusion-v1-5"
 
-pipe = AutoPipelineForText2Image.from_pretrained(
+pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float32
 )
@@ -14,19 +14,23 @@ pipe.to("cpu")
 def generate(prompt):
     image = pipe(
         prompt,
-        num_inference_steps=1,
-        guidance_scale=0.0,
         width=512,
-        height=512
+        height=512,
+        num_inference_steps=30,
+        guidance_scale=7.5
     ).images[0]
 
     return image
 
-demo = gr.Interface(
+app = gr.Interface(
     fn=generate,
-    inputs=gr.Textbox(label="Prompt"),
-    outputs=gr.Image(label="Image"),
-    title="Fast CPU AI Image Generator"
+    inputs=gr.Textbox(
+        label="Describe your image"
+    ),
+    outputs=gr.Image(
+        label="Generated Image"
+    ),
+    title="Dr Data Skeleton Image Generator"
 )
 
-demo.launch()
+app.launch()