Update app.py

app.py

@@ -10,12 +10,34 @@ model = YOLO('best.pt')
 #Function for making predictions
 def predict (image):
     results = model(image)
+    ## list of all the probabilities
+    conf = []
     for result in results:
         im_array = result.plot()
-        im = Image.fromarray(im_array[..., ::-1])  # RGB PIL image
-        # im.show()  # show image
-        # im.save('results.jpg')
-    return im
+        img = Image.fromarray(im_array[..., ::-1])  # RGB PIL image
+        img.show()
+        ### Getting the confidence
+        box = result.boxes
+        
+        conf.append(box.conf.numpy())
+
+   
+    ## sort the confidence from lowest to highest
+    if len(conf[0])>1:
+        conf_sort = sorted(conf[0]) ## ascending order
+        impression = f"Pneumothorax is detected in the image with confidence ranging from {round(conf_sort[0]*100,1)}% to {round(conf_sort[-1]*100,1)}%"
+    elif len(conf[0])==1:
+        impression = f"Pneumothorax is detected in the image with confidence of {round(conf[0][0]*100,1)}%"
+    else:
+        impression = "No pneumothorax is detected in the image"
+
+
+       
+    # print(impression)
+
+            
+    return img, impression
+        
         
 
 
@@ -23,7 +45,7 @@ def predict (image):
 platform = gr.Interface( fn = predict, 
                         title ="PTCADx: Computer-Aided Detection of Pneumothorax in Chest X-ray Images",
                         inputs = "image", 
-                        outputs = "image",
+                        outputs = ["image", "text"],
                         description="""
                         Introducing a revolutionary computer-aided detection tool designed to enhance the efficiency of clinicians in detecting pneumothorax in chest X-ray images. 
                         """,