Create generate function

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
+from diffusers import DiffusionPipeline
 
-def greet(name):
-  return "Hello " + name + "!"
+def generate(prompt, negative_prompt):
+    pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+    return pipeline(prompt, negative_prompt).images[0]
+
+iface = gr.Interface(fn=generate, inputs="text", outputs="image")
 iface.launch()

requirements.txt

@@ -0,0 +1 @@
+diffusers