Update README.md

README.md

@@ -25,7 +25,7 @@ Once installed, you can use Conda to manage environments.
 
 ### 2. **Create a New Environment**
 
-After installing Conda (https://conda.io/projects/conda/en/latest/user-guide/install/index.html), follow these steps to create a new environment and install the required packages.
+After installing Conda, follow these steps to create a new environment and install the required packages.
 
 #### **Step 1: Create a new environment**
 
@@ -179,9 +179,9 @@ from lwm_model import lwm
 import torch
 
 preprocessed_chs = tokenizer(
-    selected_scenario_names=selected_scenario_names, 
-    manual_data=None,
-    gen_raw=True
+    selected_scenario_names=selected_scenario_names,  # Selects predefined DeepMIMOv3 scenarios. Set to None to load your own dataset.
+    manual_data=None,  # If using a custom dataset, ensure it is a wireless channel dataset of size (N,32,32) based on the settings provided above.
+    gen_raw=True  # Set gen_raw=False to apply masked channel modeling (MCM), as used in LWM pre-training. For inference, masking is unnecessary unless you want to evaluate LWM's ability to handle noisy inputs.
 )
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'