Update app.py

app.py

@@ -1,36 +1,40 @@
 import gradio as gr
 import PIL.Image as Image
-
 from ultralytics import ASSETS, YOLO
 
 model = YOLO("https://huggingface.co/spaces/gpbhupinder/test/blob/main/model_-%2023%20june%202024%2019_22.pt")
 
-
 def predict_image(img):
     """Predicts objects in an image using a YOLOv8 model."""
     results = model.predict(
         source=img,
-        
         show_labels=True,
         show_conf=True,
         imgsz=640,
     )
-
+    
     for r in results:
         im_array = r.plot()
         im = Image.fromarray(im_array[..., ::-1])
-
-    return im
-
+        
+        # Extract class names and confidences
+        class_names = [model.names[int(cls)] for cls in r.boxes.cls]
+        confidences = r.boxes.conf.tolist()
+        
+        # Create a formatted string of results
+        result_text = "\n".join([f"{name}: {conf:.2f}" for name, conf in zip(class_names, confidences)])
+        
+    return im, result_text
 
 iface = gr.Interface(
     fn=predict_image,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        # gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
-        # gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold"),
     ],
-    outputs=gr.Image(type="pil", label="Result"),
+    outputs=[
+        gr.Image(type="pil", label="Result"),
+        gr.Textbox(label="Predictions")
+    ],
     title="GP Wolf Classifier",
     description="Upload images for inference.",
     examples=[