Update app.py

app.py

@@ -1,23 +1,20 @@
-# app.py
 import gradio as gr
 from transformers import pipeline
 
-# Initialize the model (example with a text-to-image model)
+# Initialize your model here
 generator = pipeline("image-generation", model="CompVis/stable-diffusion-v1-4")
 
-# Function to generate an image from a prompt
+# Function to generate image from text prompt
 def generate_image(prompt):
-    # The pipeline will return a list with the image as the first element
     return generator(prompt)[0]["image"]
 
-# Set up the Gradio interface
+# Define Gradio Interface
 interface = gr.Interface(
-    fn=generate_image,   # function to be called
-    inputs="text",       # input type: text box for user input (prompt)
-    outputs="image",     # output type: image display
-    live=True            # allows real-time generation (optional)
+    fn=generate_image,
+    inputs="text",  # User will input text prompt
+    outputs="image",  # The output will be an image
+    live=True
 )
 
 # Launch the interface
-if __name__ == "__main__":
-    interface.launch()
+interface.launch()