theme and disk utils

.streamlit/config.toml

@@ -0,0 +1,2 @@
+[theme]
+primaryColor="#feab56"

app.py

@@ -1,8 +1,14 @@
+import shutil
 import streamlit as st
 
-x = st.slider("Select a value")
-st.write(x, "squared is", x * x)
+model = st.text_input("Model name")
 
-b = st.button("Button")
+with st.container():
+    interence_mode = st.radio("Inference mode", ("causallm", "seq2seq"))
+b = st.button("Convert model")
 if b:
-    st.write("button clicked")
+    total, used, free = shutil.disk_usage("/")
+    st.write("The model name is: ", model)
+    st.write("Total: %d GiB" % (total // (2**30)))
+    st.write("Used: %d GiB" % (used // (2**30)))
+    st.write("Free: %d GiB" % (free // (2**30)))