Update app.py

app.py

@@ -677,9 +677,13 @@ 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 and datasets](https://huggingface.co/wi-lab/lwm) and follow the instructions below:
+        """)
+        gr.Markdown("""
+            <div class="explanation-box">
+                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:
+            </div>
+            
+            ```python
             from input_preprocess import DeepMIMO_data_gen deepmimo_data_cleaning label_gen # Import required modules from the model repository
             import numpy as np
             scenario_names = np.array([