Update app.py

app.py

@@ -25,7 +25,7 @@ def detect_fruit(image):
 
     # проверка на наличие детекций
     if len(detections) == 0:
-        return image, None
+        return image_cv, None
 
     result_np_image = detections[0].plot()
     result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
@@ -69,9 +69,9 @@ def gradio_interface(image, weight):
         return None, None
 
     result_np_image, detected_fruit = detect_fruit(image)
-    if not detected_fruit:
+    if len(detections) == 0:
         gr.Info('Не удалось определить товар')
-        return image, None
+        return None, None
         
     receipt = create_receipt(detected_fruit, weight)
     return result_np_image, receipt