Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ model = load_from_checkpoint(model_path)
|
|
| 8 |
def score_translation(src_text, mt_text):
|
| 9 |
translations = [{"src": src_text, "mt": mt_text}]
|
| 10 |
results = model.predict(translations, batch_size=1, gpus=1)
|
| 11 |
-
return results[
|
| 12 |
|
| 13 |
examples = [
|
| 14 |
["The weather is beautiful today.", "วันนี้อากาศดีมาก"],
|
|
@@ -17,21 +17,25 @@ examples = [
|
|
| 17 |
["Can you help me find the nearest train station?", "คุณช่วยฉันหาสถานีรถไฟที่ใกล้ที่สุดได้ไหม"]
|
| 18 |
]
|
| 19 |
|
| 20 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
gr.Markdown("# ComeTH Translation Quality Evaluator")
|
| 22 |
-
|
| 23 |
with gr.Row():
|
| 24 |
with gr.Column():
|
| 25 |
src_input = gr.Textbox(label="Source Text (English)", placeholder="Enter English text here...")
|
| 26 |
mt_input = gr.Textbox(label="Candidate Translation (Thai)", placeholder="Enter Thai translation here...")
|
| 27 |
score_button = gr.Button("Evaluate Translation", variant="primary")
|
| 28 |
-
|
| 29 |
with gr.Column():
|
| 30 |
score_output = gr.Label(label="Quality Scores")
|
| 31 |
gr.Markdown("### Higher scores indicate better translation quality across multiple dimensions")
|
| 32 |
-
|
| 33 |
gr.Examples(examples=examples, inputs=[src_input, mt_input], outputs=score_output, fn=score_translation)
|
| 34 |
-
|
| 35 |
score_button.click(fn=score_translation, inputs=[src_input, mt_input], outputs=score_output)
|
| 36 |
|
| 37 |
if __name__ == "__main__":
|
|
|
|
| 8 |
def score_translation(src_text, mt_text):
|
| 9 |
translations = [{"src": src_text, "mt": mt_text}]
|
| 10 |
results = model.predict(translations, batch_size=1, gpus=1)
|
| 11 |
+
return results["scores"][0]
|
| 12 |
|
| 13 |
examples = [
|
| 14 |
["The weather is beautiful today.", "วันนี้อากาศดีมาก"],
|
|
|
|
| 17 |
["Can you help me find the nearest train station?", "คุณช่วยฉันหาสถานีรถไฟที่ใกล้ที่สุดได้ไหม"]
|
| 18 |
]
|
| 19 |
|
| 20 |
+
font_css = """
|
| 21 |
+
@import url("https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;700&display=swap");
|
| 22 |
+
|
| 23 |
+
* {
|
| 24 |
+
font-family: 'JetBrains Mono', monospace !important;
|
| 25 |
+
}
|
| 26 |
+
"""
|
| 27 |
+
|
| 28 |
+
with gr.Blocks(theme=gr.themes.Soft(), css=font_css) as demo:
|
| 29 |
gr.Markdown("# ComeTH Translation Quality Evaluator")
|
|
|
|
| 30 |
with gr.Row():
|
| 31 |
with gr.Column():
|
| 32 |
src_input = gr.Textbox(label="Source Text (English)", placeholder="Enter English text here...")
|
| 33 |
mt_input = gr.Textbox(label="Candidate Translation (Thai)", placeholder="Enter Thai translation here...")
|
| 34 |
score_button = gr.Button("Evaluate Translation", variant="primary")
|
|
|
|
| 35 |
with gr.Column():
|
| 36 |
score_output = gr.Label(label="Quality Scores")
|
| 37 |
gr.Markdown("### Higher scores indicate better translation quality across multiple dimensions")
|
|
|
|
| 38 |
gr.Examples(examples=examples, inputs=[src_input, mt_input], outputs=score_output, fn=score_translation)
|
|
|
|
| 39 |
score_button.click(fn=score_translation, inputs=[src_input, mt_input], outputs=score_output)
|
| 40 |
|
| 41 |
if __name__ == "__main__":
|