Update app.py

Trying to get the list working

app.py

@@ -66,12 +66,15 @@ def detect_objects(model_name,url_input,image_input,threshold):
     elif 'yolos' in model_name:
     
         model = YolosForObjectDetection.from_pretrained(model_name)
-    
+
+    tb_label = ""
     if validators.url(url_input):
         image = Image.open(requests.get(url_input, stream=True).raw)
+        tb_label = "Confidence Values URL"
         
     elif image_input:
         image = image_input
+        tb_label = "Confidence Values Upload"
     
     #Make prediction
     processed_output_list = make_prediction(image, feature_extractor, model)
@@ -82,8 +85,8 @@ def detect_objects(model_name,url_input,image_input,threshold):
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
     
     # return [viz_img, processed_outputs]
-    print(type(viz_img))
-    return viz_img
+    # print(type(viz_img))
+    return viz_img, gr.Textbox(value=str(processed_outputs),label=tb_label)
         
 def set_example_image(example: list) -> dict:
     return gr.Image.update(value=example[0])
@@ -150,16 +153,15 @@ with demo:
                 
             img_but = gr.Button('Detect')
 
-    output_text1 = gr.Textbox(value="", label="Confidence Values URL")
-    output_text2 = gr.Textbox(value="", label="Confidence Values Upload")
+    output_text1 = gr.Textbox(value="", label="Confidence Values")
     
-    # 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)
+    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_text1],queue=True)
     # 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,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,queue=True)
+    # img_but.click(detect_objects,inputs=[options,url_input,img_input,slider_input],outputs=img_output_from_upload,queue=True)
 
     
     example_images.click(fn=set_example_image,inputs=[example_images],outputs=[img_input])