Update app.py

app.py

@@ -196,7 +196,8 @@ def create_gradio_interface():
                 layer_name = gr.Dropdown(
                     label="Select Layer",
                     choices=[name for name in model.named_modules() if 'layer' in name or 'embeddings' in name],
-                    value="embeddings"
+                    value="embeddings",
+                    allow_custom_value=True
                 )
                 visualize_option = gr.Radio(
                     label="Visualization Type",