Update app.py

app.py

@@ -47,8 +47,8 @@ def yolov8_img_inference(
     # image = read_image
     results = model.predict(image)
     render = render_result(model=model, image=image, result=results[0])
-    count_result = results[0].boxes.cls.tolist().count('name')
-    # count_result = results.pandas().xyxy[0].value_counts('name')
+    # count_result = results[0].boxes.cls.tolist().count('name')
+    count_result = results.pandas().xyxy[0].value_counts('name')
     return render, count_result
     
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
@@ -93,15 +93,15 @@ inputs_image = [
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
 ]
 
-# outputs_image =gr.outputs.Image(type="filepath", label="Output Image")
-# count_obj = gr.outputs.Textbox(show_label=True)
+outputs_image =gr.outputs.Image(type="filepath", label="Output Image")
+count_obj = gr.outputs.Textbox(show_label=False)
 
 title = "Tất cả do anh Đạt"
 
 interface_image = gr.Interface(
     fn=yolov8_img_inference,
     inputs=inputs_image,
-    outputs=[gr.Image(type="pil"),gr.Textbox(show_label=False)],
+    outputs=[outputs_image, count_obj],
     title=title,
     examples=image_path,
     cache_examples=True,