Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
-hf_writer = gr.HuggingFaceDatasetSaver("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", "diffusiondb")
+from diffusers import DiffusionPipeline
 
-def greet(name):
-    return hf_writer
+pipeline = DiffusionPipeline.from_pretrained("lucasfelezdev/TextToImage")
+
+def greet(text):
+    image = pipeline("text").images[0]
+    return image
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="image")