MrFeelgoood commited on
Commit
4ebe9a5
1 Parent(s): fec4160

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -251,14 +251,14 @@ def pdf_ocr(file, model_t, question):
251
 
252
  return immobile_values, total_output, text_output
253
 
254
- def ocr_interface(pdf_file, model_t, question):
 
255
  # Call the pdf_ocr function
256
  values, total, text = pdf_ocr(pdf_file.name, model_t, question)
257
  return values, total, text
258
 
259
 
260
-
261
-
262
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
263
 
264
  gr.Markdown(
@@ -294,7 +294,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
294
  extract_button.click(fn = ocr_interface,
295
  inputs=[pdf_input, model_input, question_input], outputs=[values_output, total_output, text_output])
296
 
297
- gr.Examples(['Example1(scanned).pdf', 'Example2.pdf', 'Example3Large.pdf'], inputs = pdf_input,
298
  cache_examples = True, fn = ocr_interface, outputs = [values_output, total_output, text_output])
299
 
300
  demo.launch()
 
251
 
252
  return immobile_values, total_output, text_output
253
 
254
+
255
+ def ocr_interface(pdf_file, model_t='it5/it5-base-question-answering', question="Quanti metri quadri misura l'immobile?"):
256
  # Call the pdf_ocr function
257
  values, total, text = pdf_ocr(pdf_file.name, model_t, question)
258
  return values, total, text
259
 
260
 
261
+ # Start the UI
 
262
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
263
 
264
  gr.Markdown(
 
294
  extract_button.click(fn = ocr_interface,
295
  inputs=[pdf_input, model_input, question_input], outputs=[values_output, total_output, text_output])
296
 
297
+ gr.Examples(['Example1(scannedDoc).pdf', 'Example2.pdf', 'Example3Large.pdf'], inputs = pdf_input,
298
  cache_examples = True, fn = ocr_interface, outputs = [values_output, total_output, text_output])
299
 
300
  demo.launch()