Update app.py

app.py

@@ -61,13 +61,17 @@ def predict(image, prompt):
     num_images_per_prompt=1,
   ).images
 
+  draw_on_image = ImageDraw.Draw(image)
+  # Define the rectangle coordinates (left-top, right-bottom)
+  rectangle_coordinates = coords
+  draw_on_image.rectangle(rectangle_coordinates, outline="red", width=2)
 
-  return(images[0])
+  return images[0], image
     
 examples = [["cats.png", "cat is smiling"],
             ["dog.jpg", "dog with big eyes"],
             ["dog1.jpg", "dog with big bone"],
-            ["limurs.jpg", "limur eating bug"]]
+            ["limurs.jpg", "limur with bug in mouth"]]
 
 gr.Interface(
     predict,