Update README.md

README.md

@@ -50,4 +50,13 @@ In order to force transcription for a certain language set the 1. decoder id as
         )
         return {"sequences": outputs["sequences"]}
 
+    def translate(self, input_features):
+        outputs = self.model.generate(
+            input_features,
+            max_new_tokens=450,  # change as needed
+            return_dict_in_generate=True,
+            forced_decoder_ids=[[1, 50261], [2, 50358], [3, 50363]],  # different forced_decoder_ids
+        )
+        return {"sequences": outputs["sequences"]}
+
 (language codes from here: https://github.com/woheller69/whisperIME/blob/master/app/src/main/java/com/whispertflite/utils/InputLang.java)