Update README.md

README.md

@@ -11,7 +11,7 @@ tags:
 - xlsr-fine-tuning-week
 license: apache-2.0
 model-index:
-- name: wav2vec2-large-xlsr-mongolian-v1
+- name: XLSR Wav2Vec2 Mongolian V1 by Bayartsogt
   results:
   - task: 
       name: Speech Recognition
@@ -23,7 +23,7 @@ model-index:
     metrics:
        - name: Test WER
          type: wer
-         value: 35.42
+         value: 35.33
 ---
 
 # Wav2Vec2-Large-XLSR-53-Mongolian-v1
@@ -72,7 +72,7 @@ def evaluate(batch):
      with torch.no_grad():
           logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
 
-     pred_ids = torch.argmax(logits, dim=-
+     pred_ids = torch.argmax(logits, dim=-1)
      batch["pred_strings"] = processor.batch_decode(pred_ids)
      return batch
 
@@ -81,7 +81,7 @@ result = test_dataset.map(evaluate, batched=True, batch_size=8)
 print("WER: {:2f}".format(100 * wer.compute(predictions=result["pred_strings"], references=result["sentence"])))
 ```
 
-**Test Result**: 35.42 % 
+**Test Result**: 35.33 % 
 
 
 ## Training