File size: 445 Bytes
562ef99
c892105
 
562ef99
db5635b
562ef99
 
 
db5635b
562ef99
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)))