Update app.py

app.py

@@ -33,8 +33,6 @@ def detect_fruit(image):
     detected_fruit = None
 
     for det in detections:
-        print('kekkekek')
-        print(det.boxes.cls)
         label = model.names[int(det.boxes.cls)]  # название фрукта
         if label in fruits_data:
             detected_fruit = label
@@ -44,7 +42,7 @@ def detect_fruit(image):
 
 # =========================== ЧЕК ============================
 def create_receipt(detected_fruit, weight):
-    data = fruits_data[detected_fruit];
+    data = fruits_data[detected_fruit]
     fruit_name = data['name']
     price = data['price_per_kg']
     total_price = round(price * weight, 2)
@@ -62,10 +60,7 @@ def create_receipt(detected_fruit, weight):
     draw.text((10, 110), f"Цена за кг: {price} руб.", fill="black", font=font)
     draw.text((10, 140), f"Сумма: {total_price} руб.", fill="black", font=font)
 
-    with BytesIO() as output:
-        receipt_img.save(output, format="PNG")
-        output.seek(0)
-        return output.getvalue()
+    return receipt_img
 
 # ======================= ИНТЕРФЕЙС ============================
 def gradio_interface(image, weight):