Tsunnami commited on
Commit
4a19d1f
·
verified ·
1 Parent(s): 0785af4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -10,13 +10,22 @@ def score_translation(src_text, 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.", "วันนี้อากาศดีมาก"],
15
  ["I need to go to the hospital.", "ฉันต้องไปโรงพยาบาล"],
16
  ["This restaurant serves delicious food.", "ร้านอาหารนี้เสิร์ฟอาหารอร่อย"],
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
 
@@ -35,7 +44,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=font_css) as demo:
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__":
 
10
  results = model.predict(translations, batch_size=1, gpus=1)
11
  return results["scores"][0]
12
 
13
+ good_examples = [
14
  ["The weather is beautiful today.", "วันนี้อากาศดีมาก"],
15
  ["I need to go to the hospital.", "ฉันต้องไปโรงพยาบาล"],
16
  ["This restaurant serves delicious food.", "ร้านอาหารนี้เสิร์ฟอาหารอร่อย"],
17
  ["Can you help me find the nearest train station?", "คุณช่วยฉันหาสถานีรถไฟที่ใกล้ที่สุดได้ไหม"]
18
  ]
19
 
20
+ bad_examples = [
21
+ ["The weather is beautiful today.", "วันนี้อากาศแย่สุดๆ"],
22
+ ["I need to go to the hospital.", "ฉันอยากกินข้าว"],
23
+ ["This restaurant serves delicious food.", "ร้านนี้ไม่อร่อยเลย"],
24
+ ["Can you help me find the nearest train station?", "คุณพูดภาษาอังกฤษได้ไหม?"],
25
+ ]
26
+
27
+ examples = good_examples + bad_examples
28
+
29
  font_css = """
30
  @import url("https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;700&display=swap");
31
 
 
44
  with gr.Column():
45
  score_output = gr.Label(label="Quality Scores")
46
  gr.Markdown("### Higher scores indicate better translation quality across multiple dimensions")
47
+ gr.Examples(
48
+ examples=examples,
49
+ inputs=[src_input, mt_input],
50
+ outputs=score_output,
51
+ fn=score_translation
52
+ )
53
  score_button.click(fn=score_translation, inputs=[src_input, mt_input], outputs=score_output)
54
 
55
  if __name__ == "__main__":