Update src/streamlit_app.py

src/streamlit_app.py

@@ -97,7 +97,6 @@ DEFAULT_CORPUS = [
 #Select from 4 different models
 MODELS = ["Qwen/Qwen1.5-0.5B", "deepseek-ai/deepseek-coder-1.3b-instruct", "openai-community/gpt2", "prem-research/MiniGuard-v0.1"]
 
-"""## Defining Utility Functions"""
 
 # ====== Utilities =========================================================================
 def seed_everything(seed: int = 42):
@@ -628,7 +627,7 @@ def main():
         st.header("Graph")
         graph_mode = st.selectbox("Graph mode", ["knn", "threshold"], index=0)
         knn_k = st.slider("k (kNN)", 2, 50, 8) if graph_mode == "knn" else 8
-        sim_threshold = st.slider("Similarity threshold", 0.0, 0.99, 0.70, step=0.01) if graph_mode == "threshold" else 0.70
+        sim_threshold = st.slider("Similarity threshold", 0.0, 0.99, 0.05, step=0.01) if graph_mode == "threshold" else 0.70
         use_cosine = st.checkbox("Use cosine similarity", value=True)
 
         st.header("Anchors / LoT")
@@ -648,18 +647,18 @@ def main():
 
         prompt_col, run_col = st.columns([4, 1])
 
-        with prompt_col:
-            main_text = st.selectbox(
-                "Prompt to visualize (hidden states computed on this text)",
-                options=DEFAULT_CORPUS,
-                index=0,
-                help="Select a predefined prompt for analysis"
-            )
+    with prompt_col:
+        main_text = st.selectbox(
+            "Prompt to visualize (hidden states computed on this text)",
+            options=DEFAULT_CORPUS,
+            index=0,
+            help="Select a predefined prompt for analysis"
+        )
 
-        with run_col:
-            st.write("")
-            st.write("")
-            run_btn = st.button("Run", type="primary")
+    with run_col:
+        st.write("")
+        st.write("")
+        run_btn = st.button("Run", type="primary")
 
     cfg = Config(
         model_name=model_name,