Update app.py

app.py

@@ -35,7 +35,7 @@ def fig2img(fig):
     return img
 
 
-def visualize_prediction(pil_img, output_dict, threshold=0.7, id2label=None):
+def visualize_prediction(pil_img, output_dict, threshold=0.5, id2label=None):
     keep = output_dict["scores"] > threshold
     boxes = output_dict["boxes"][keep].tolist()
     scores = output_dict["scores"][keep].tolist()
@@ -121,7 +121,7 @@ with demo:
         with gr.TabItem('Image URL'):
             with gr.Row():
                 url_input = gr.Textbox(lines=2,label='Enter valid image URL here..')
-                img_output_from_url = gr.Image(shape=(650,650))
+                img_output_from_url = gr.Image(shape=(750,750))
                 
             with gr.Row():
                 example_url = gr.Dataset(components=[url_input],samples=[[str(url)] for url in urls])
@@ -135,7 +135,7 @@ with demo:
                 
             with gr.Row(): 
                 example_images = gr.Dataset(components=[img_input],
-                                            samples=[[path.as_posix()]
+                                            samples=[[path.as_posix("images")]
                                                      for path in sorted(pathlib.Path('images').rglob('*.JPG'))])
                 
             img_but = gr.Button('Detect')