aehrm commited on
Commit
d2673e5
·
1 Parent(s): a3620e9
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,9 +25,9 @@ def predict(input_str, model_name, progress=gr.Progress()):
25
 
26
  if type(output_sentences[0]) == list:
27
  detok = TreebankWordDetokenizer()
28
- return "<br>".join([detok.detokenize(recombine_tokens(sent)) for sent in output_sentences])
29
  else:
30
- return "<br>".join(output_sentences)
31
 
32
  def predict_transnormer(tokenized_sentences, progress):
33
  model_name = 'ybracke/transnormer-19c-beta-v02'
@@ -117,8 +117,8 @@ def predict_type_transformer_with_lm(tokenized_sentences, progress):
117
 
118
  gradio_app = gr.Interface(
119
  predict,
120
- inputs=[gr.Textbox(value="Die Königinn ſaß auf des Pallaſtes mittlerer Tribune."), gr.Dropdown([('aehrm/dtaec-type-normalizer (FAST)', 'type normalizer'), ('aehrm/dtaec-type-normalizer + dbmdz/german-gpt2 (Fast)', 'type normalizer + lm'), ('ybracke/transnormer-19c-beta-v02 (fast)', 'transnormer')])],
121
- outputs=gr.HTML(),
122
  title="German Historical Text Normalization",
123
  )
124
 
 
25
 
26
  if type(output_sentences[0]) == list:
27
  detok = TreebankWordDetokenizer()
28
+ return "\n".join([detok.detokenize(recombine_tokens(sent)) for sent in output_sentences])
29
  else:
30
+ return "\n".join(output_sentences)
31
 
32
  def predict_transnormer(tokenized_sentences, progress):
33
  model_name = 'ybracke/transnormer-19c-beta-v02'
 
117
 
118
  gradio_app = gr.Interface(
119
  predict,
120
+ inputs=[gr.Textbox(label="Input", value="Die Königinn ſaß auf des Pallaſtes mittlerer Tribune."), gr.Dropdown([('aehrm/dtaec-type-normalizer (FAST)', 'type normalizer'), ('aehrm/dtaec-type-normalizer + dbmdz/german-gpt2 (Fast)', 'type normalizer + lm'), ('ybracke/transnormer-19c-beta-v02 (fast)', 'transnormer')], label="Model")],
121
+ outputs=gr.Textbox(label="Output", show_label=True),
122
  title="German Historical Text Normalization",
123
  )
124