Update app.py

app.py

@@ -120,7 +120,7 @@ lang_list = list(lang_ids.keys())
 
 def translate(lang, text):
     encoded = tokenizer(text, return_tensors="pt")
-    generated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id(lang_ids[lang]))
+    generated_tokens = model(use_cache=True).generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id(lang_ids[lang]))
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]