Spaces:
Running
on
Zero
Running
on
Zero
update app.py, fix-bug, return text_result
Browse files
app.py
CHANGED
|
@@ -103,7 +103,7 @@ def process_image(image, model_size, task_type, is_eval_mode):
|
|
| 103 |
|
| 104 |
# Return all three results. Gradio will handle the temporary file path for the image.
|
| 105 |
text_result = plain_text_result if plain_text_result else markdown_content
|
| 106 |
-
return result_image, markdown_content,
|
| 107 |
|
| 108 |
|
| 109 |
# Create Gradio interface
|
|
@@ -172,8 +172,8 @@ with gr.Blocks(title="DeepSeek-OCR", theme=gr.themes.Soft()) as demo:
|
|
| 172 |
["examples/receipt.jpg", "Base", "Convert to Markdown"],
|
| 173 |
["examples/receipt-2.png", "Base", "Convert to Markdown"],
|
| 174 |
],
|
| 175 |
-
inputs=[image_input, model_size, task_type],
|
| 176 |
-
outputs=[output_image, output_markdown, output_text
|
| 177 |
fn=process_image,
|
| 178 |
cache_examples=False,
|
| 179 |
)
|
|
|
|
| 103 |
|
| 104 |
# Return all three results. Gradio will handle the temporary file path for the image.
|
| 105 |
text_result = plain_text_result if plain_text_result else markdown_content
|
| 106 |
+
return result_image, markdown_content, text_result
|
| 107 |
|
| 108 |
|
| 109 |
# Create Gradio interface
|
|
|
|
| 172 |
["examples/receipt.jpg", "Base", "Convert to Markdown"],
|
| 173 |
["examples/receipt-2.png", "Base", "Convert to Markdown"],
|
| 174 |
],
|
| 175 |
+
inputs=[image_input, model_size, task_type, eval_mode_checkbox],
|
| 176 |
+
outputs=[output_image, output_markdown, output_text],
|
| 177 |
fn=process_image,
|
| 178 |
cache_examples=False,
|
| 179 |
)
|