Update README.md

README.md

@@ -41,10 +41,10 @@ from transformers import AutoModel
 model = AutoModel.from_pretrained('oriyonay/myna-85m')
 
 # Myna supports unbatched (2D) and batched (3D or 4D) inputs:
-output = model(torch.randn(128, 96))  # shape (1, 384)
-output = model(torch.randn(2, 128, 96))  # shape (2, 384)
-output = model(torch.randn(2, 1, 128, 96))  # shape (2, 384)
+output = model(torch.randn(128, 96))  # shape (1, 1536)
+output = model(torch.randn(2, 128, 96))  # shape (2, 1536)
+output = model(torch.randn(2, 1, 128, 96))  # shape (2, 1536)
 
 # Additionally, you can load audio directly from a file:
-output = model.from_file('your_file.wav')  # shape (n_chunks, 384)
+output = model.from_file('your_file.wav')  # shape (n_chunks, 1536)
 ```