alphahg commited on
Commit
cf06234
1 Parent(s): f42eb22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -13,17 +13,12 @@ import openai
13
  openai.api_key = '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'
14
  gpt2_tokenizer = AutoTokenizer.from_pretrained('gpt2')
15
 
16
- import os
17
- # if not(os.environ['JAVA_HOME']):
18
- # os.environ['JAVA_HOME'] = r'C:\Program Files\Java\jdk-17\bin\server'
19
-
20
  #en2ko = 'alphahg/m2m100_418M-finetuned-en-to-ko-4770260'#'alphahg/mbart-large-50-finetuned-en-to-ko-8603428-finetuned-en-to-ko-9914408'
21
  en2ko = 'alphahg/mbart-large-50-finetuned-en-to-ko-8603428-finetuned-en-to-ko-9914408'
22
  ko2en = 'alphahg/opus-mt-ko-en-finetuned-ko-to-en-2780616'
23
  ensum = 'allenai/led-large-16384-arxiv'
24
  kosum = 'alphahg/pko-t5-small-finetuned-paper-4564652' #'lcw99/t5-base-korean-text-summary'
25
 
26
- kkma = Kkma()
27
  #en_pipe = pipeline('translation', model=en2ko, tokenizer=en2ko, src_lang = "en", tgt_lang = "ko", device_map="auto")
28
  en2ko_model = AutoModelForSeq2SeqLM.from_pretrained(en2ko)
29
 
 
13
  openai.api_key = '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'
14
  gpt2_tokenizer = AutoTokenizer.from_pretrained('gpt2')
15
 
 
 
 
 
16
  #en2ko = 'alphahg/m2m100_418M-finetuned-en-to-ko-4770260'#'alphahg/mbart-large-50-finetuned-en-to-ko-8603428-finetuned-en-to-ko-9914408'
17
  en2ko = 'alphahg/mbart-large-50-finetuned-en-to-ko-8603428-finetuned-en-to-ko-9914408'
18
  ko2en = 'alphahg/opus-mt-ko-en-finetuned-ko-to-en-2780616'
19
  ensum = 'allenai/led-large-16384-arxiv'
20
  kosum = 'alphahg/pko-t5-small-finetuned-paper-4564652' #'lcw99/t5-base-korean-text-summary'
21
 
 
22
  #en_pipe = pipeline('translation', model=en2ko, tokenizer=en2ko, src_lang = "en", tgt_lang = "ko", device_map="auto")
23
  en2ko_model = AutoModelForSeq2SeqLM.from_pretrained(en2ko)
24