Update app.py

app.py

@@ -185,18 +185,17 @@ def check_object_in_image(query_image, target_image, threshold=50, scale_factor=
 
     return {
         'is_present': is_present,
-        'images': images_to_return,
-        'object detection inliers': [int(i) for i in cropped_inliers],
-        'bbox_image': bbox_image,
+        'image_with_boxes': bbox_image,
+        'object_detection_inliers': [int(i) for i in cropped_inliers],
     }
 
 def interface(poster_source, media_source, threshold, scale_factor):
     result1 = check_object_in_image(poster_source, media_source, threshold, scale_factor)
     if result1['is_present']:
-        return result1
+        return result1['is_present'], result1['image_with_boxes']
 
     result2 = check_object_in_image(poster_source, media_source, threshold, scale_factor)
-    return result2 if result2['is_present'] else result1
+    return result2['is_present'], result2['image_with_boxes']
 
 iface = gr.Interface(
     fn=interface,