polish fix
Browse files
app.py
CHANGED
|
@@ -103,6 +103,8 @@ async def translate(request: Request):
|
|
| 103 |
chunk = f">>nob<< {chunk}"
|
| 104 |
if model_id == "Helsinki-NLP/opus-mt-tc-base-en-sh":
|
| 105 |
chunk = f">>cnr<< {chunk}"
|
|
|
|
|
|
|
| 106 |
|
| 107 |
# tokenize
|
| 108 |
inputs = tokenizer(chunk, return_tensors="pt", padding=True, truncation=True)
|
|
|
|
| 103 |
chunk = f">>nob<< {chunk}"
|
| 104 |
if model_id == "Helsinki-NLP/opus-mt-tc-base-en-sh":
|
| 105 |
chunk = f">>cnr<< {chunk}"
|
| 106 |
+
if model_id == "Helsinki-NLP/opus-mt-en-sla":
|
| 107 |
+
chunk = f">>pol<< {chunk}"
|
| 108 |
|
| 109 |
# tokenize
|
| 110 |
inputs = tokenizer(chunk, return_tensors="pt", padding=True, truncation=True)
|