alphahg commited on
Commit
0e6379c
1 Parent(s): fd135ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ kosum = 'alphahg/pko-t5-small-finetuned-paper-4564652' #'lcw99/t5-base-korean-te
23
  en2ko_model = AutoModelForSeq2SeqLM.from_pretrained(en2ko)
24
 
25
  en_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "en_XX", tgt_lang = "ko_KR")
26
- ko_pipe = pipeline('translation', model=ko2en, tokenizer=ko2en, device="cuda:0")
27
  style_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "ko_KR", tgt_lang = "ko_KR")
28
 
29
  en_sum = pipeline('summarization', model=ensum, tokenizer=ensum)
 
23
  en2ko_model = AutoModelForSeq2SeqLM.from_pretrained(en2ko)
24
 
25
  en_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "en_XX", tgt_lang = "ko_KR")
26
+ ko_pipe = pipeline('translation', model=ko2en, tokenizer=ko2en)
27
  style_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "ko_KR", tgt_lang = "ko_KR")
28
 
29
  en_sum = pipeline('summarization', model=ensum, tokenizer=ensum)