henok3878 commited on
Commit
15b9e77
·
1 Parent(s): cc28e40

add model training dataset details and usage notes

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,7 +36,8 @@ def translate(text: str, beam_size: int = 4) -> str:
36
  with gr.Blocks(title="Transformer From Scratch Translation Demo") as demo:
37
  gr.Markdown(
38
  "# Transformer From Scratch Translation Demo\n"
39
- "Translate English to German using a custom Transformer model trained from scratch."
 
40
  )
41
  with gr.Row(equal_height=True):
42
  with gr.Column():
 
36
  with gr.Blocks(title="Transformer From Scratch Translation Demo") as demo:
37
  gr.Markdown(
38
  "# Transformer From Scratch Translation Demo\n"
39
+ "Translate English to German using a custom Transformer model trained from scratch.\n\n"
40
+ "**Note:** This model was trained on the WMT14 English-German news dataset. It works best on formal, news-style sentences and may not perform well on everyday informal or conversational text."
41
  )
42
  with gr.Row(equal_height=True):
43
  with gr.Column():