Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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="/
|
| 10 |
-
model_sk_en = MarianMTModel.from_pretrained(model_name_sk_en, cache_dir="/
|
| 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="/
|
| 16 |
-
model_en_sk = MarianMTModel.from_pretrained(model_name_en_sk, cache_dir="/
|
| 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):
|