Update app.py

app.py

@@ -3,6 +3,7 @@ import random
 from PIL import Image
 import io
 
+# Define a function to generate an image from a prompt
 def generate_image(prompt):
     try:
         # Generate a random seed for each request
@@ -11,11 +12,10 @@ def generate_image(prompt):
         # Generate image with the model
         image = model.generate(prompt, seed=random_seed)
         
-        # Check if the image is a PIL Image, if not try to convert
+        # Ensure the image is in PIL format
         if isinstance(image, bytes):
             image = Image.open(io.BytesIO(image))
         elif not isinstance(image, Image.Image):
-            # If image is not in PIL format, convert or handle it
             return "Error: Image not in expected format"
         
         return image
@@ -27,4 +27,6 @@ model = gr.load("models/black-forest-labs/FLUX.1-schnell")
 
 # Create Gradio interface
 iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
+
+# Launch the Gradio interface
 iface.launch()