Update app.py

app.py

@@ -53,8 +53,8 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
     ax = plt.gca()
     colors = COLORS * 100
     for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
-        ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=5))
-        ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=30, bbox=dict(facecolor="yellow", alpha=0.5))
+        ax.add_patch(plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, color=color, linewidth=8))
+        ax.text(xmin, ymin, f"{label}: {score:0.2f}", fontsize=40, bbox=dict(facecolor="yellow", alpha=0.5))
     plt.axis("off")
     return fig2img(plt.gcf())
     
@@ -127,7 +127,7 @@ with demo:
         gr.Markdown(title)
         gr.Markdown(description)
         gr.Markdown(twitter_link)
-        options = gr.Dropdown(choices=models,label='Object Detection Model',show_label=True)
+        options = gr.Dropdown(choices=models,label='Object Detection Model',value=models[0],show_label=True)
         slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.5,step=0.1,label='Prediction Threshold')
         
         with gr.Tabs():