Update app.py

app.py

@@ -675,26 +675,22 @@ with gr.Blocks(css="""
                 hdf.create_dataset('labels', data=labels)
             print("Dataset saved!")
 
+
             # To use your preferred DeepMIMO scenarios for the custom dataset, please [clone the model and datasets](https://huggingface.co/wi-lab/lwm) and follow the instructions below:
-            from input_preprocess import DeepMIMO_data_gen deepmimo_data_cleaning label_gen
-            
+            from input_preprocess import DeepMIMO_data_gen deepmimo_data_cleaning label_gen # Import required modules from the model repository
             scenario_names = np.array([
                 "city_18_denver", "city_15_indianapolis", "city_19_oklahoma", 
                 "city_12_fortworth", "city_11_santaclara", "city_7_sandiego"
             ])
-            scenario_name = scenario_names[0]
-            
-            deepmimo_data = DeepMIMO_data_gen(scenario_name)
-            cleaned_deepmimo_data = deepmimo_data_cleaning(deepmimo_data)
+            scenario_name = scenario_names[0] # Select the scenario by choosing its index.
+            deepmimo_data = DeepMIMO_data_gen(scenario_name) # Generates ray-traced wireless channels for the selected scenario.
+            cleaned_deepmimo_data = deepmimo_data_cleaning(deepmimo_data) # Filters out users with no direct path to the base station (i.e., users with zero-valued channels).
             
-            channels = np.squeeze(np.array(cleaned_deepmimo_data), axis=1)
-            labels = label_gen('LoS/NLoS Classification', deepmimo_data, scenario_name)
+            channels = np.squeeze(np.array(cleaned_deepmimo_data), axis=1) # The "channels" array is now prepared for packing into the custom dataset in H5 format.
+            labels = label_gen('LoS/NLoS Classification', deepmimo_data, scenario_name) # Generates labels for each user, classifying them as Line-of-Sight (LoS) or Non-Line-of-Sight (NLoS), and prepares the "labels" array for inclusion in the custom dataset H5 file.
             ```
         """)
 
-
-        
-
 # Launch the app
 if __name__ == "__main__":
     demo.launch()