hashing troubles(

app.py

@@ -3,8 +3,8 @@ import streamlit as st
 from transformers import pipeline
 
 @st.cache
-def get_model():
-	return pipeline("fill-mask", model="roberta-base")
+def get_model(model):
+	return pipeline("fill-mask", model=model)
 
 
 text = st.text_input("Enter a sentence. Use a * for a mask.")
@@ -13,7 +13,7 @@ text = st.text_input("Enter a sentence. Use a * for a mask.")
 
 if text:
 	data_load_state = st.text('Loading data...')
-	nlp = get_model()
+	nlp = get_model("roberta-base")
 	result = nlp(text.replace("*",nlp.tokenizer.mask_token))
 	data_load_state.text('Loading data...done!')
 	for d in result: