fix ui

app.py

@@ -2,6 +2,7 @@ import random
 
 import numpy as np
 import streamlit as st
+import nltk
 import torch
 import umap
 from nltk.tokenize import word_tokenize
@@ -19,8 +20,6 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 torch.manual_seed(42)
 np.random.seed(42)
 random.seed(42)
-import nltk
-
 nltk.download("punkt")
 
 
@@ -63,12 +62,16 @@ def main():
         11,
         value=9,
     )
-    is_centering = st.sidebar.checkbox("centering embeddings", value=True)
-    ot_type = st.sidebar.selectbox(
-        "ot_type", ["POT", "UOT", "OT"], help="optimal transport algorithm to be used"
+    is_centering = st.sidebar.toggle("centering embeddings", value=True)
+    ot_type = st.sidebar.radio(
+        "type",
+        ["OT", "POT", "UOT"],
+        index=1,
+        horizontal=True,
+        help="optimal transport algorithm to be used",
     )
     ot_type = ot_type.lower()
-    sinkhorn = st.sidebar.checkbox(
+    sinkhorn = st.sidebar.toggle(
         "sinkhorn", value=True, help="use sinkhorn algorithm"
     )
     distortion = st.sidebar.slider(
@@ -92,7 +95,7 @@ def main():
         value=0.22,
         help="sparsity of alignment matrix",
     )
-    show_assignments = st.sidebar.checkbox("show assignments", value=True)
+    show_assignments = st.sidebar.toggle("show assignments", value=True)
     if show_assignments:
         n_neighbors = st.sidebar.slider(
             "n_neighbors", 2, 10, value=8, help="number of neighbors for umap"