Update app.py

Trying to see if I can get the json

app.py

@@ -24,7 +24,7 @@ def make_prediction(img, feature_extractor, model):
     outputs = model(**inputs)
     img_size = torch.tensor([tuple(reversed(img.size))])
     processed_outputs = feature_extractor.post_process(outputs, img_size)
-    return processed_outputs[0]
+    return processed_outputs
 
 def fig2img(fig):
     buf = io.BytesIO()
@@ -72,7 +72,9 @@ def detect_objects(model_name,url_input,image_input,threshold):
         image = image_input
     
     #Make prediction
-    processed_outputs = make_prediction(image, feature_extractor, model)
+    processed_output_list = make_prediction(image, feature_extractor, model)
+    print(processed_output_list[1:])
+    processed_output = processed_output_list[0]
     
     #Visualize prediction
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
@@ -101,9 +103,9 @@ Links to HuggingFace Models:
 models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small','hustvl/yolos-tiny']
 urls = ["https://c8.alamy.com/comp/J2AB4K/the-new-york-stock-exchange-on-the-wall-street-in-new-york-J2AB4K.jpg"]
 
-twitter_link = """
-[![](https://img.shields.io/twitter/follow/nickmuchi?label=@nickmuchi&style=social)](https://twitter.com/nickmuchi)
-"""
+# twitter_link = """
+# [![](https://img.shields.io/twitter/follow/nickmuchi?label=@nickmuchi&style=social)](https://twitter.com/nickmuchi)
+# """
 
 css = '''
 h1#title {
@@ -115,7 +117,7 @@ demo = gr.Blocks(css=css)
 with demo:
     gr.Markdown(title)
     gr.Markdown(description)
-    gr.Markdown(twitter_link)
+    # 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')