axiilay commited on
Commit
6001782
·
1 Parent(s): aff0c0d

update app.py, fix-bug, return text_result

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, plain_text_result
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, eval_mode_checkbox],
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
  )