update ui
Browse files
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 "
|
29 |
else:
|
30 |
-
return "
|
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.
|
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 |
|