Update README.md

README.md

@@ -1,13 +1,18 @@
 ---
 license: mit
+language:
+- en
+- ru
 ---
 
-# Convert MUSE from TensorFlow to PyTorch and ONNX
+# Convert MUSE from TensorFlow to PyTorch
 
-Read more about the project: [GitHub](https://github.com/dayyass/muse_tf2pt/tree/main).
+This repository contains code to use mUSE (Multilingual Universal Sentence Encoder) transformer model from [TF Hub](https://www.kaggle.com/models/google/universal-sentence-encoder/tensorFlow2/multilingual-large) using **PyTorch**.
 
 > [!IMPORTANT]
-> **The PyTorch model can be used not only for inference, but also for additional training and fine-tuning**.
+> **The PyTorch model can be used not only for inference, but also for additional training and fine-tuning!**
+
+Read more about the project: [GitHub](https://github.com/dayyass/muse_tf2pt/tree/main).
 
 # Usage
 
@@ -17,8 +22,8 @@ Model initialization and usage code:
 ```python
 import torch
 from functools import partial
-from src.architecture import MUSE
-from src.tokenizer import get_tokenizer, tokenize
+from architecture import MUSE
+from tokenizer import get_tokenizer, tokenize
 
 PATH_TO_PT_MODEL = "model.pt"
 PATH_TO_TF_MODEL = "universal-sentence-encoder-multilingual-large-3"