Update app.py

app.py

@@ -91,8 +91,11 @@ def detect_and_crop(target_image, query_image, threshold=0.5, nms_threshold=0.3)
     return filtered_boxes, target_image
 
 def image_matching_no_pyramid(query_img, target_img, visualize=True):
-    image1, inp1, scales1 = read_image(target_img, device, [640*2], 0, True)
-    image0, inp0, scales0 = read_image(query_img, device, [640*2], 0, True)
+    temp_query = save_array_to_temp_image(np.array(query_img))
+    temp_target = save_array_to_temp_image(np.array(target_img))
+    
+    image1, inp1, scales1 = read_image(temp_target, device, [640*2], 0, True)
+    image0, inp0, scales0 = read_image(temp_query, device, [640*2], 0, True)
 
     if image0 is None or image1 is None:
         return None
@@ -137,15 +140,16 @@ def check_object_in_image(query_image, target_image, threshold=50, scale_factor=
     cropped_images, bbox_image = detect_and_crop(target_image, query_image)
 
     temp_files = [save_array_to_temp_image(i) for i in cropped_images]
-    crop_results = [image_matching_no_pyramid(query_image, i, visualize=True) for i in temp_files]
+    crop_results = [image_matching_no_pyramid(query_image, Image.open(i), visualize=True) for i in temp_files]
 
     cropped_visuals = []
     cropped_inliers = []
     for result in crop_results:
-        for img in result['visualized_image']:
-            cropped_visuals.append(Image.fromarray(img))
-        for inliers_ in result['inliers']:
-            cropped_inliers.append(inliers_)
+        if result:
+            for img in result['visualized_image']:
+                cropped_visuals.append(Image.fromarray(img))
+            for inliers_ in result['inliers']:
+                cropped_inliers.append(inliers_)
 
     images_to_return.append(stitch_images(cropped_visuals))