Update README.md

README.md

@@ -282,10 +282,10 @@ assert sr == 16000, "Input of wav2vec2.0 is expected to have sampling rate of 16
 input_values = processor(audio, sampling_rate=16000, return_tensors="pt").cuda()
 
 # translation to German
-emb, mask = zeroswot_encoder(**input_values)
+compressed_embeds, attention_mask = zeroswot_encoder(**input_values)
 predicted_ids = nllb_model.generate(
-    inputs_embeds=emb,
-    attention_mask=~mask,
+    inputs_embeds=compressed_embeds,
+    attention_mask=attention_mask,
     forced_bos_token_id=tokenizer.lang_code_to_id["deu_Latn"],
     num_beams=5,
 )