Update app.py

app.py

@@ -76,15 +76,17 @@ def predict(img) :
     pred = plot_img_bbox(np.array(img), nms_prediction)
 
     #pred = np.array(Image.open("pred.jpg"))
+    word = "Number of palm trees detected : "+str(len(nms_prediction["boxes"]))
     
     # Calculate the prediction time
     pred_time = round(timer() - start_time, 5)
     
     # Return the prediction dictionary and prediction time 
-    return pred
+    return pred,word
     
 image = gr.components.Image()
 out_im = gr.components.Image()
+out_lab = gr.components.Label()
 
 ### 4. Gradio app ###
 # Create title, description and article strings
@@ -100,7 +102,7 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
 # Create the Gradio demo
 demo = gr.Interface(fn=predict, # mapping function from input to output
                     inputs= image, #gr.Image(type="pil"), # what are the inputs?
-                    outputs=out_im,
+                    outputs=[out_im,out_lab]
                     examples=example_list, 
                     title=title,
                     description=description,