Update app.py

app.py

@@ -22,12 +22,12 @@ kosum = 'alphahg/pko-t5-small-finetuned-paper-4564652' #'lcw99/t5-base-korean-te
 #en_pipe = pipeline('translation', model=en2ko, tokenizer=en2ko, src_lang = "en", tgt_lang = "ko", device_map="auto")
 en2ko_model = AutoModelForSeq2SeqLM.from_pretrained(en2ko)
 
-en_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "en_XX", tgt_lang = "ko_KR", device="cuda:0")
+en_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "en_XX", tgt_lang = "ko_KR")
 ko_pipe = pipeline('translation', model=ko2en, tokenizer=ko2en, device="cuda:0")
-style_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "ko_KR", tgt_lang = "ko_KR", device="cuda:0")
+style_pipe = pipeline('translation', model=en2ko_model, tokenizer=en2ko, src_lang = "ko_KR", tgt_lang = "ko_KR")
 
-en_sum = pipeline('summarization', model=ensum, tokenizer=ensum, device="cuda:1")
-ko_sum = pipeline('summarization', model=kosum, tokenizer=kosum, device="cuda:1")
+en_sum = pipeline('summarization', model=ensum, tokenizer=ensum)
+ko_sum = pipeline('summarization', model=kosum, tokenizer=kosum)
 
 def len_tokens(text, pipe):
     return len(pipe.tokenizer(text)['input_ids'])