Update app.py

app.py

@@ -9,14 +9,12 @@ pipeline = pipeline.to("cuda" if torch.cuda.is_available() else "cpu")
 def generate_image(prompt):
     # Generate an image based on the prompt using the pipeline
     output = pipeline(prompt)
-    # Assuming that the output has a `images` field with a list of images
     return output.images[0]
 
-# Create Gradio interface
+# Gradio interface setup
 title = "Shap-E Model Demo"
 description = "Generate images from text prompts using the Shap-E diffusion model."
 
-# Set up the interface
 interface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Prompt"),
@@ -25,6 +23,5 @@ interface = gr.Interface(
     description=description,
 )
 
-# Launch the Gradio app
 if __name__ == "__main__":
     interface.launch()