Update app.py

app.py

@@ -25,7 +25,7 @@ def detect_fruit(image):
 
     # Проверка на наличие детекций
     if len(detections) == 0:
-        return image
+        return image, None
 
     result_np_image = detections[0].plot()
     result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
@@ -33,7 +33,9 @@ def detect_fruit(image):
     detected_fruit = None
 
     for det in detections:
-        label = model.names[9]  # название фрукта
+        print('kekkekek')
+        print(det.boxes.cls)
+        label = model.names[int(det.boxes.cls)]  # название фрукта
         if label in fruits_data:
             detected_fruit = label
             break
@@ -93,7 +95,7 @@ image_output = gr.Image(
 )
 receipt_output = gr.Image(
     label="Чек",
-    type="numpy",
+    type="auto",
     width=640,
     height=480
 )