hayanaka commited on
Commit
efb718c
1 Parent(s): 99ab353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -3,16 +3,16 @@ import gradio as gr
3
 
4
  # Pipeline
5
  # Translate from Japanese to English
6
- translator_ja_to_en = pipeline("translation_ja_to_en", model="Helsinki-NLP/opus-mt-ja-en")
7
- # translator_ja_to_en = pipeline("translation_ja_to_en", model="japanese-denim/nllb-finetuned-naga-to-eng")
8
 
9
  # Summerize from English to English
10
  # summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
11
  summarizer = pipeline("summarization", model="google/pegasus-large")
12
 
13
  # Translate from English to Japanese
14
- translator_en_to_ja = pipeline("translation_en_to_ja", model="Helsinki-NLP/opus-tatoeba-en-ja")
15
- # translator_en_to_ja = pipeline("translation_en_to_ja", model="ZenXir/marian-finetuned-kde4-en-to-ja")
16
 
17
  maxlength = 500
18
 
@@ -30,16 +30,14 @@ def summarize_and_translate(text):
30
  # Translate from English to Japanese
31
  summary_in_ja = translator_en_to_ja(summary_in_en, max_length=100)[0]['translation_text']
32
 
33
- # summary_in_ja = re.sub(r'\s+', ' ', summary_in_ja).strip()
34
-
35
  return summary_in_ja
36
 
37
  demo = gr.Interface(
38
  fn=summarize_and_translate,
39
- inputs=gr.Textbox(lines=5, placeholder= f"{maxlength}文字以内で日本語の文章を入力してください\nPlease enter Japanese text within a {maxlength}-character limit"),
40
  outputs=gr.Textbox(lines=5, label="日本語要約 Summarized Japanese"),
41
  title="日本語要約 Jpanese Summarizer",
42
- description=f"日本語の文章を入力すると、日本語の要約が表示されます。\n Enter Japanese text up to {maxlength} characters to see its summary."
43
  )
44
 
45
  demo.launch()
 
3
 
4
  # Pipeline
5
  # Translate from Japanese to English
6
+ # translator_ja_to_en = pipeline("translation_ja_to_en", model="Helsinki-NLP/opus-mt-ja-en")
7
+ translator_ja_to_en = pipeline("translation_ja_to_en", model="japanese-denim/nllb-finetuned-naga-to-eng")
8
 
9
  # Summerize from English to English
10
  # summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
11
  summarizer = pipeline("summarization", model="google/pegasus-large")
12
 
13
  # Translate from English to Japanese
14
+ # translator_en_to_ja = pipeline("translation_en_to_ja", model="Helsinki-NLP/opus-tatoeba-en-ja")
15
+ translator_en_to_ja = pipeline("translation_en_to_ja", model="ZenXir/marian-finetuned-kde4-en-to-ja")
16
 
17
  maxlength = 500
18
 
 
30
  # Translate from English to Japanese
31
  summary_in_ja = translator_en_to_ja(summary_in_en, max_length=100)[0]['translation_text']
32
 
 
 
33
  return summary_in_ja
34
 
35
  demo = gr.Interface(
36
  fn=summarize_and_translate,
37
+ inputs=gr.Textbox(lines=5, placeholder= f"{maxlength}文字以内で日本語の文章を入力してください Please enter Japanese text within a {maxlength}-character limit"),
38
  outputs=gr.Textbox(lines=5, label="日本語要約 Summarized Japanese"),
39
  title="日本語要約 Jpanese Summarizer",
40
+ description=f"日本語の文章を入力すると、日本語の要約が表示されます。 Enter Japanese text up to {maxlength} characters to see its summary."
41
  )
42
 
43
  demo.launch()