Spaces:
Runtime error
Runtime error
Ankur Goyal
commited on
Commit
·
15fad86
1
Parent(s):
8393858
Hide the outputs by default
Browse files
app.py
CHANGED
|
@@ -89,17 +89,29 @@ def process_path(path):
|
|
| 89 |
if path:
|
| 90 |
try:
|
| 91 |
document = load_document(path)
|
| 92 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
except Exception:
|
| 94 |
pass
|
| 95 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
|
| 97 |
|
| 98 |
def process_upload(file):
|
| 99 |
if file:
|
| 100 |
return process_path(file.name)
|
| 101 |
else:
|
| 102 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
| 103 |
|
| 104 |
|
| 105 |
colors = ["#64A087", "green", "black"]
|
|
@@ -126,7 +138,9 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
|
|
| 126 |
)
|
| 127 |
draw.rectangle(((x1, y1), (x2, y2)), fill=(0, 255, 0, int(0.4 * 255)))
|
| 128 |
|
| 129 |
-
return image,
|
|
|
|
|
|
|
| 130 |
|
| 131 |
|
| 132 |
def load_example_document(img, question, model):
|
|
@@ -183,15 +197,21 @@ with gr.Blocks(css=CSS) as demo:
|
|
| 183 |
submit_button = gr.Button("Submit", variant="primary", elem_id="submit-button")
|
| 184 |
|
| 185 |
with gr.Row():
|
| 186 |
-
image = gr.Image(visible=
|
| 187 |
with gr.Column():
|
| 188 |
-
output = gr.JSON(label="Output")
|
| 189 |
|
| 190 |
clear_button.click(
|
| 191 |
-
lambda _: (
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 192 |
inputs=clear_button,
|
| 193 |
outputs=[image, document, question, output],
|
| 194 |
)
|
|
|
|
| 195 |
upload.change(fn=process_upload, inputs=[upload], outputs=[document, image, output])
|
| 196 |
url.change(fn=process_path, inputs=[url], outputs=[document, image, output])
|
| 197 |
|
|
|
|
| 89 |
if path:
|
| 90 |
try:
|
| 91 |
document = load_document(path)
|
| 92 |
+
return (
|
| 93 |
+
document,
|
| 94 |
+
gr.update(visible=True, value=document.preview),
|
| 95 |
+
gr.update(visible=False, value=None),
|
| 96 |
+
)
|
| 97 |
except Exception:
|
| 98 |
pass
|
| 99 |
+
return (
|
| 100 |
+
None,
|
| 101 |
+
gr.update(visible=False, value=None),
|
| 102 |
+
gr.update(visible=False, value=None),
|
| 103 |
+
)
|
| 104 |
|
| 105 |
|
| 106 |
def process_upload(file):
|
| 107 |
if file:
|
| 108 |
return process_path(file.name)
|
| 109 |
else:
|
| 110 |
+
return (
|
| 111 |
+
None,
|
| 112 |
+
gr.update(visible=False, value=None),
|
| 113 |
+
gr.update(visible=False, value=None),
|
| 114 |
+
)
|
| 115 |
|
| 116 |
|
| 117 |
colors = ["#64A087", "green", "black"]
|
|
|
|
| 138 |
)
|
| 139 |
draw.rectangle(((x1, y1), (x2, y2)), fill=(0, 255, 0, int(0.4 * 255)))
|
| 140 |
|
| 141 |
+
return gr.update(visible=True, value=image), gr.update(
|
| 142 |
+
visible=True, value=predictions
|
| 143 |
+
)
|
| 144 |
|
| 145 |
|
| 146 |
def load_example_document(img, question, model):
|
|
|
|
| 197 |
submit_button = gr.Button("Submit", variant="primary", elem_id="submit-button")
|
| 198 |
|
| 199 |
with gr.Row():
|
| 200 |
+
image = gr.Image(visible=False)
|
| 201 |
with gr.Column():
|
| 202 |
+
output = gr.JSON(label="Output", visible=False)
|
| 203 |
|
| 204 |
clear_button.click(
|
| 205 |
+
lambda _: (
|
| 206 |
+
gr.update(visible=False, value=None),
|
| 207 |
+
None,
|
| 208 |
+
None,
|
| 209 |
+
gr.update(visible=False, value=None),
|
| 210 |
+
),
|
| 211 |
inputs=clear_button,
|
| 212 |
outputs=[image, document, question, output],
|
| 213 |
)
|
| 214 |
+
|
| 215 |
upload.change(fn=process_upload, inputs=[upload], outputs=[document, image, output])
|
| 216 |
url.change(fn=process_path, inputs=[url], outputs=[document, image, output])
|
| 217 |
|