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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
 
@@ -25,10 +25,10 @@ def summarize_and_translate(text):
25
  translated_text_to_en = translator_ja_to_en(text, max_length=maxlength+500)[0]['translation_text']
26
 
27
  # Summerize from English to English
28
- summary_in_en = summarizer(translated_text_to_en, min_length=100, max_length=300)[0]['summary_text']
29
 
30
  # Translate from English to Japanese
31
- summary_in_ja = translator_en_to_ja(summary_in_en, max_length=400)[0]['translation_text']
32
 
33
  # summary_in_ja = re.sub(r'\s+', ' ', summary_in_ja).strip()
34
 
 
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
 
 
25
  translated_text_to_en = translator_ja_to_en(text, max_length=maxlength+500)[0]['translation_text']
26
 
27
  # Summerize from English to English
28
+ summary_in_en = summarizer(translated_text_to_en, min_length=50, max_length=200)[0]['summary_text']
29
 
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