Update app.py

app.py

@@ -674,6 +674,17 @@ with gr.Blocks(css="""
                 hdf.create_dataset('channels', data=channels)
                 hdf.create_dataset('labels', data=labels)
             print("Dataset saved!")
+
+            # To use your preferred DeepMIMO scenarios for the custom dataset, please [clone the model](https://huggingface.co/wi-lab/lwm) and follow the instructions below:
+            from input_preprocess import DeepMIMO_data_gen deepmimo_data_cleaning label_gen
+            
+            scenario_names = ['city_19_oklahoma']
+            
+            deepmimo_data = [DeepMIMO_data_gen(scenario_name) for scenario_name in scenario_names]
+            cleaned_deepmimo_data = [deepmimo_data_cleaning(deepmimo_data[scenario_idx]) for scenario_idx in range(len(scenario_names))]
+            
+            channels = np.squeeze(np.array(cleaned_deepmimo_data[0]), axis=1)
+            labels = label_gen('LoS/NLoS Classification', deepmimo_data[0], scenario_names, n_beams=64)
             ```
         """)