Update app.py

app.py

@@ -14,7 +14,8 @@ def get_model(model):
 @st.cache(allow_output_mutation=True)
 def loading_models(model='roberta-base'):
      return get_model(model), SentenceTransformer('all-MiniLM-L6-v2')
-
+     
+@st.cache(allow_output_mutation=True)
 def main(nlp, semantic_model, data_load_state):
     data_load_state.text('Inference from model...')
     result = nlp(text+' '+nlp.tokenizer.mask_token)