Update app.py

Trying to get confidences from image.

app.py

@@ -42,7 +42,7 @@ def visualize_prediction(pil_img, output_dict, threshold=0.7, id2label=None):
     if id2label is not None:
         labels = [id2label[x] for x in labels]
 
-    print("Labels " + str(labels))
+    # print("Labels " + str(labels))
 
     plt.figure(figsize=(16, 10))
     plt.imshow(pil_img)
@@ -81,7 +81,7 @@ def detect_objects(model_name,url_input,image_input,threshold):
     #Visualize prediction
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
     
-    return viz_img   
+    return viz_img, processed_outputs
         
 def set_example_image(example: list) -> dict:
     return gr.Image.update(value=example[0])
@@ -122,6 +122,9 @@ with demo:
     # gr.Markdown(twitter_link)
     options = gr.Dropdown(choices=models,label='Select Object Detection Model',show_label=True)
     slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.7,label='Prediction Threshold')
+
+    output_text1 = gr.Textbox(value="", label="Confidence Values URL")
+    output_text2 = gr.Textbox(value="", label="Confidence Values Upload")
     
     with gr.Tabs():
         with gr.TabItem('Image URL'):
@@ -147,13 +150,13 @@ with demo:
             img_but = gr.Button('Detect')
         
     
-    url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_url,queue=True)
-    img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_upload,queue=True)
+    url_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_url, output_text1],queue=True)
+    img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=[img_output_from_upload, output_text2],queue=True)
     example_images.click(fn=set_example_image,inputs=[example_images],outputs=[img_input])
     example_url.click(fn=set_example_url,inputs=[example_url],outputs=[url_input])
     
 
-    gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-object-detection-with-detr-and-yolos)")
+    # gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-object-detection-with-detr-and-yolos)")
 
     
 demo.launch(enable_queue=True)