Update app.py

app.py

@@ -1,13 +1,10 @@
-import gradio as gr
+import gradio as gr 
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 import torch
-from diffusers import StableDiffusionPipeline
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
-
-def diffusion(prompt, filename):
-    image = pipe(prompt).images[0]
-    image.save(filename)
-    return image
-
-iface = gr.Interface(fn=diffusion, inputs="text", outputs="image")
-iface.launch()
+def txt2img(prompt):
+  image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0]
+  # image.save("sd_image.png")
+  return image
+    
+gr.Interface(txt2img, gr.Text(), gr.Image(), title = 'Stable Diffusion 2.0 Colab with Gradio UI').launch(share = True, debug = True)