Update app.py

app.py

@@ -9,7 +9,7 @@ model = YOLO('best.pt')
 
 #Function for making predictions
 def predict (image):
-    results = model(image)
+    results = model(image, conf =0.2)
     ## list of all the probabilities
     conf = []
     for result in results:
@@ -45,7 +45,10 @@ def predict (image):
 platform = gr.Interface( fn = predict, 
                         title ="PTCADx: Computer-Aided Detection of Pneumothorax in Chest X-ray Images",
                         inputs = "image", 
-                        outputs = ["image", "text"],
+                        outputs = [
+                                gr.outputs.Image(label="Processed Image"),
+                                gr.outputs.Text(label="Impression")
+                            ],
                         description="""
                         Introducing a revolutionary computer-aided detection tool designed to enhance the efficiency of clinicians in detecting pneumothorax in chest X-ray images. 
                         """,