import shutil
import streamlit as st

model = st.text_input("Model name")

with st.container():
    interence_mode = st.radio("Inference mode", ("causallm", "seq2seq"))
b = st.button("Convert model")
if b:
    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)))