Update app.py

app.py

@@ -16,7 +16,7 @@ def make_prediction(img, feature_extractor, model):
     inputs = feature_extractor(img, return_tensors="pt")
     outputs = model(**inputs)
     img_size = torch.tensor([tuple(reversed(img.size))])
-    processed_outputs = feature_extractor.post_process(outputs, img_size)
+    processed_outputs = feature_extractor.post_process_object_detection(outputs, img_size)
     print(outputs.keys())
     return (
         processed_outputs[0],
@@ -75,7 +75,6 @@ def set_example_image(example: list):
 
 with gr.Blocks(css=css) as app:
     gr.Markdown(title)
-    gr.Markdown(description)
 
     with gr.Tabs():
         with gr.TabItem("Image upload and detections visualization"):
@@ -130,6 +129,9 @@ with gr.Blocks(css=css) as app:
         with gr.TabItem("Model details"):
             with gr.Row():
                 model_details = gr.Markdown(""" """)
+        with gr.TabItem("Dataset details"):
+            with gr.Row():
+                gr.Markdown(description)
 
     detect_button.click(
         detect_objects,