try
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def process_image(image, text_input=None):
|
|
| 53 |
image = Image.fromarray(image) # Convert NumPy array to PIL Image
|
| 54 |
task_prompt = '<DocVQA>'
|
| 55 |
results = run_example(task_prompt, image, text_input)[task_prompt].replace("<pad>", "")
|
| 56 |
-
return results
|
| 57 |
|
| 58 |
|
| 59 |
css = """
|
|
@@ -74,7 +74,6 @@ with gr.Blocks(css=css) as demo:
|
|
| 74 |
submit_btn = gr.Button(value="Submit")
|
| 75 |
with gr.Column():
|
| 76 |
output_text = gr.Textbox(label="Output Text")
|
| 77 |
-
output_img = gr.Image(label="Output Image")
|
| 78 |
|
| 79 |
gr.Examples(
|
| 80 |
examples=[
|
|
@@ -84,12 +83,12 @@ with gr.Blocks(css=css) as demo:
|
|
| 84 |
["image.jpg", "What's the share of Industry Switchers Gained?"]
|
| 85 |
],
|
| 86 |
inputs=[input_img, text_input],
|
| 87 |
-
outputs=[output_text
|
| 88 |
fn=process_image,
|
| 89 |
cache_examples=True,
|
| 90 |
label='Try examples'
|
| 91 |
)
|
| 92 |
|
| 93 |
-
submit_btn.click(process_image, [input_img, text_input], [output_text
|
| 94 |
|
| 95 |
demo.launch(debug=True)
|
|
|
|
| 53 |
image = Image.fromarray(image) # Convert NumPy array to PIL Image
|
| 54 |
task_prompt = '<DocVQA>'
|
| 55 |
results = run_example(task_prompt, image, text_input)[task_prompt].replace("<pad>", "")
|
| 56 |
+
return results
|
| 57 |
|
| 58 |
|
| 59 |
css = """
|
|
|
|
| 74 |
submit_btn = gr.Button(value="Submit")
|
| 75 |
with gr.Column():
|
| 76 |
output_text = gr.Textbox(label="Output Text")
|
|
|
|
| 77 |
|
| 78 |
gr.Examples(
|
| 79 |
examples=[
|
|
|
|
| 83 |
["image.jpg", "What's the share of Industry Switchers Gained?"]
|
| 84 |
],
|
| 85 |
inputs=[input_img, text_input],
|
| 86 |
+
outputs=[output_text],
|
| 87 |
fn=process_image,
|
| 88 |
cache_examples=True,
|
| 89 |
label='Try examples'
|
| 90 |
)
|
| 91 |
|
| 92 |
+
submit_btn.click(process_image, [input_img, text_input], [output_text])
|
| 93 |
|
| 94 |
demo.launch(debug=True)
|