Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -80,17 +80,23 @@ def run_grounding(input_image, grounding_caption, box_threshold, text_threshold)
|
|
| 80 |
return image_with_box
|
| 81 |
|
| 82 |
if __name__ == "__main__":
|
| 83 |
-
|
| 84 |
parser = argparse.ArgumentParser("Grounding DINO demo", add_help=True)
|
| 85 |
parser.add_argument("--debug", action="store_true", help="using debug mode")
|
| 86 |
parser.add_argument("--share", action="store_true", help="share the app")
|
| 87 |
args = parser.parse_args()
|
| 88 |
-
|
| 89 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
with block:
|
| 91 |
-
gr.Markdown("
|
| 92 |
-
gr.Markdown("
|
| 93 |
-
gr.Markdown("Note the model runs on CPU, so it may take a while to run the model
|
| 94 |
|
| 95 |
with gr.Row():
|
| 96 |
with gr.Column():
|
|
@@ -115,6 +121,13 @@ if __name__ == "__main__":
|
|
| 115 |
|
| 116 |
run_button.click(fn=run_grounding, inputs=[
|
| 117 |
input_image, grounding_caption, box_threshold, text_threshold], outputs=[gallery])
|
| 118 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
block.launch(share=False, show_api=False, show_error=True)
|
| 120 |
|
|
|
|
| 80 |
return image_with_box
|
| 81 |
|
| 82 |
if __name__ == "__main__":
|
| 83 |
+
|
| 84 |
parser = argparse.ArgumentParser("Grounding DINO demo", add_help=True)
|
| 85 |
parser.add_argument("--debug", action="store_true", help="using debug mode")
|
| 86 |
parser.add_argument("--share", action="store_true", help="share the app")
|
| 87 |
args = parser.parse_args()
|
| 88 |
+
css = """
|
| 89 |
+
#mkd {
|
| 90 |
+
height: 500px;
|
| 91 |
+
overflow: auto;
|
| 92 |
+
border: 1px solid #ccc;
|
| 93 |
+
}
|
| 94 |
+
"""
|
| 95 |
+
block = gr.Blocks(css=css).queue()
|
| 96 |
with block:
|
| 97 |
+
gr.Markdown("<h1><center>Grounding DINO<h1><center>")
|
| 98 |
+
gr.Markdown("<h3><center>Open-World Detection with <a href='https://github.com/IDEA-Research/GroundingDINO'>Grounding DINO</a><h3><center>")
|
| 99 |
+
gr.Markdown("<h3><center>Note the model runs on CPU, so it may take a while to run the model.<h3><center>")
|
| 100 |
|
| 101 |
with gr.Row():
|
| 102 |
with gr.Column():
|
|
|
|
| 121 |
|
| 122 |
run_button.click(fn=run_grounding, inputs=[
|
| 123 |
input_image, grounding_caption, box_threshold, text_threshold], outputs=[gallery])
|
| 124 |
+
gr.Examples(
|
| 125 |
+
[["this_is_fine.png", "coffee cup", 0.25, 0.25]],
|
| 126 |
+
inputs = [input_image, grounding_caption, box_threshold, text_threshold],
|
| 127 |
+
outputs = [gallery],
|
| 128 |
+
fn=run_grounding,
|
| 129 |
+
cache_examples=True,
|
| 130 |
+
label='Try this example input!'
|
| 131 |
+
)
|
| 132 |
block.launch(share=False, show_api=False, show_error=True)
|
| 133 |
|