deboker commited on
Commit
4952563
Β·
verified Β·
1 Parent(s): e76ae4c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -6,14 +6,14 @@ app = FastAPI()
6
 
7
  # SK β†’ EN
8
  model_name_sk_en = "Helsinki-NLP/opus-mt-sk-en"
9
- tokenizer_sk_en = MarianTokenizer.from_pretrained(model_name_sk_en, cache_dir="/tmp")
10
- model_sk_en = MarianMTModel.from_pretrained(model_name_sk_en, cache_dir="/tmp")
11
  sk_to_en = pipeline("translation", model=model_sk_en, tokenizer=tokenizer_sk_en)
12
 
13
  # EN β†’ SK
14
  model_name_en_sk = "Helsinki-NLP/opus-mt-en-sk"
15
- tokenizer_en_sk = MarianTokenizer.from_pretrained(model_name_en_sk, cache_dir="/tmp")
16
- model_en_sk = MarianMTModel.from_pretrained(model_name_en_sk, cache_dir="/tmp")
17
  en_to_sk = pipeline("translation", model=model_en_sk, tokenizer=tokenizer_en_sk)
18
 
19
  class TranslationRequest(BaseModel):
 
6
 
7
  # SK β†’ EN
8
  model_name_sk_en = "Helsinki-NLP/opus-mt-sk-en"
9
+ tokenizer_sk_en = MarianTokenizer.from_pretrained(model_name_sk_en, cache_dir="/app/cache")
10
+ model_sk_en = MarianMTModel.from_pretrained(model_name_sk_en, cache_dir="/app/cache")
11
  sk_to_en = pipeline("translation", model=model_sk_en, tokenizer=tokenizer_sk_en)
12
 
13
  # EN β†’ SK
14
  model_name_en_sk = "Helsinki-NLP/opus-mt-en-sk"
15
+ tokenizer_en_sk = MarianTokenizer.from_pretrained(model_name_en_sk, cache_dir="/app/cache")
16
+ model_en_sk = MarianMTModel.from_pretrained(model_name_en_sk, cache_dir="/app/cache")
17
  en_to_sk = pipeline("translation", model=model_en_sk, tokenizer=tokenizer_en_sk)
18
 
19
  class TranslationRequest(BaseModel):