Update README.md

README.md

@@ -12,7 +12,7 @@ We uploaded a pre-trained model trained on the spatialized DNS challenge dataset
 
 ## Using pre-trained model
 Download `deftan-II.py` and `max.pt` to `your\path\`.
-Declare the DeFTAN-II class and load the `state_dict` using the code below.
+You may import the DeFTAN-II class and load the `state_dict` using the code below.
 
 ```
 import torch
@@ -24,7 +24,7 @@ state_dict = {k.replace("module.", ""): v for k, v in state_dict.items()}
 DeFTAN2model.load_state_dict(state_dict)
 ```
 
-You can now use the trained parameter to enhance speech.
+You can now use the DeFTAN-II with pre-trained parameter to enhance speech.
 
 ---
 license: mit