Update app.py

app.py

@@ -23,9 +23,12 @@ detections = []
 
 # =========================== ДЕТЕКЦИЯ ПЛОДА ============================
 
-def detect_fruit(image, weight):
-    results = model.predict(source=image, conf=0.5)
-    detections = results[0].boxes.data.tolist()
+def detect_image(image_path: str, weight: float):
+    image = cv2.imread(image_path)
+    detections = model.predict(source=image_path, conf=0.5)
+    result_np_image = detections[0].plot()
+    result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
+    return result_np_image
     detected_fruit = None
     
     for det in detections:
@@ -87,6 +90,6 @@ gr.Interface(
     fn=gradio_interface,
     inputs=[image_input, weight_input],
     outputs=[image_output, receipt_output],
-    title="Определение фрукта и создание чека",
-    description="Загрузите изображение фрукта на весах, введите вес и получите чек"
+    title="Определение плода и создание чека",
+    description="Загрузите изображение, введите вес и получите чек"
 ).launch()