Update app.py

app.py

@@ -36,30 +36,36 @@ def restore_version(index, versions):
 
 
 def generate(image_editor, prompt, neg_prompt, versions):
-    print('versions')
-    print(versions)
-    
     image = image_editor['background'].convert('RGB')
+
+    # Resize image
     image.thumbnail((1024, 1024))
     image = divisible_by_8(image)
     original_image_size = image.size
-    
+
+    # Mask layer
     layer = image_editor["layers"][0].resize(image.size)
-    
+
+    # Make image a square
     image = squarify_image(image)
-    
+
+    # Make sure mask is white with a black background
     mask = Image.new("RGBA", image.size, "WHITE") 
     mask.paste(layer, (0, 0), layer)
     mask = ImageOps.invert(mask.convert('L'))
 
+    # Inpaint
     final_image = pipeline(prompt=prompt, 
                            image=image, 
                            mask_image=mask).images[0]
-    
+
+    # Crop image to original aspect ratio
     final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))
 
+    # gradio.ImageEditor requires a diction
     final_dict = {'background': final_image, 'layers': None, 'composite': final_image}
 
+    # Add generated image to version gallery
     if(versions==None): 
         final_gallery = [image_editor['background'] ,final_image]
     else: 
@@ -72,6 +78,8 @@ with gr.Blocks() as demo:
     gr.Markdown("""
     # Inpainting Sketch Pad
     by [Tony Assi](https://www.tonyassi.com/)
+
+    Please ❤️ this Space. I build custom AI apps for companies. <a href="mailto: tony.assi.media@gmail.com">Email me</a> for business inquiries.
     """)
     
     with gr.Row():