jvdzwaan commited on
Commit
003cd01
·
1 Parent(s): 220e257

Improve demo interface

Browse files

by adding a title, description and a larger input text area.

Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -8,6 +8,11 @@ from ocrpostcorrection.icdar_data import generate_sentences, process_input_ocr
8
  from ocrpostcorrection.token_classification import tokenize_and_align_labels
9
  from ocrpostcorrection.utils import predictions_to_labels, predictions2entity_output
10
 
 
 
 
 
 
11
  tokenizer_name = 'bert-base-multilingual-cased'
12
  model_name = 'jvdzwaan/ocrpostcorrection-task-1'
13
 
@@ -45,8 +50,10 @@ def tag(text):
45
  examples = ['This is a cxample...']
46
 
47
  demo = gr.Interface(tag,
48
- gr.Textbox(placeholder="Enter sentence here..."),
49
  gr.HighlightedText(),
 
 
50
  examples=examples,
51
  allow_flagging='never')
52
 
 
8
  from ocrpostcorrection.token_classification import tokenize_and_align_labels
9
  from ocrpostcorrection.utils import predictions_to_labels, predictions2entity_output
10
 
11
+ description = """
12
+ Token classifier that predicts whether a token is an OCR mistake or not. More
13
+ information can be found [here](https://jvdzwaan.github.io/ocrpostcorrection/).
14
+ """
15
+
16
  tokenizer_name = 'bert-base-multilingual-cased'
17
  model_name = 'jvdzwaan/ocrpostcorrection-task-1'
18
 
 
50
  examples = ['This is a cxample...']
51
 
52
  demo = gr.Interface(tag,
53
+ gr.Textbox(placeholder="Enter sentence here...", lines=7),
54
  gr.HighlightedText(),
55
+ title='Find OCR mistakes in text',
56
+ description=description,
57
  examples=examples,
58
  allow_flagging='never')
59