Create app.py

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from transformers import DalleBartTokenizer, DalleBartForConditionalGeneration
+
+# Load the tokenizer and model
+tokenizer = DalleBartTokenizer.from_pretrained("dalle-mini/dalle-mini")
+model = DalleBartForConditionalGeneration.from_pretrained("dalle-mini/dalle-mini")
+
+def generate_image(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(**inputs)
+    image = outputs[0]  # This will depend on your model's output format
+    return image
+
+# Define Gradio interface
+iface = gr.Interface(fn=generate_image,
+                     inputs="text",
+                     outputs="image",
+                     title="DALL-E Mini Image Generator",
+                     description="Enter a prompt to generate an image.")
+
+if __name__ == "__main__":
+    iface.launch()
+
+
+
+