Update app.py

app.py

@@ -222,7 +222,7 @@ def translate(src_sentence: str, strategy:str = 'greedy' , lenght_extend :int =
     else:
         assert length_penalty >= 0 , 'lenght penelity must be superior or equal to zero'
         assert beam_size >= 1, 'beam size must superior or equal to one'
-        tgt_tokens = beam_search_decode(model, src, src_mask, maxt_len=num_tokens + lenght_extend, start_symbol=target_tokenizer.bos_token_id, beam_size=beam_size,length_penalty=length_penalty).flatten()
+        tgt_tokens = beam_search_decode(model, src, src_mask, max_len=num_tokens + lenght_extend, start_symbol=target_tokenizer.bos_token_id, beam_size=beam_size,length_penalty=length_penalty).flatten()
     # Decode the target tokens and clean up the result
     return target_tokenizer.decode(tgt_tokens, clean_up_tokenization_spaces=True, skip_special_tokens=True)