seanp03 commited on
Commit
b4d0c67
·
verified ·
1 Parent(s): eb01c52

polish fix

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)