Update app.py

app.py

@@ -14,8 +14,9 @@ pad_id = tokenizer._convert_token_to_id_with_added_voc("<pad>")
 CREOLE = {"Hawaiian Pidgin": "hwc", "Saint Lucian Creole": "acf", "Belizean Creole": "bzj", "Chavacano Creole": "cbk", "Seychellois Creole": "crs", "Sranan Tongo": "srn", "Aukan": "djk", "Gullah": "gul", "San Andrés–Providencia Creole": "icr", "Jamaican Creole": "jam", "Mauritian Creole": "mfe", "Papiamento": "pap", "Pijin": "pis", "Tok Pisin": "tpi", "Torres Strait Creole": "tcs", "Australian Kriol": "rop", "Sango": "sag", "Saramaccan": "srm", "Bislama": "bis", "Nigerian": "pcm", "Sierra Leonean Creole": "kri", "Haitian Creole": "hat", "Kupang Malay": "mkn", "Tetun Dili": "tdt", "Malay Baba": "mbf", "Kituba": "ktu"}
 
 
-def generate(input, task, slang, tlang):
-    lang = CREOLE[lang]
+def generate(input, slang, tlang):
+    slang = CREOLE[slang]
+    tlang = CREOLE[tlang]
     inp = tokenizer(input.strip() + " </s> <2" + slang + ">",
                     add_special_tokens=False, return_tensors="pt", padding=True).input_ids
     model_output = model.generate(inp, use_cache=True, num_beams=1, max_length=100, min_length=1, early_stopping=True, pad_token_id=pad_id,