Update app.py

app.py

@@ -1,3 +1,10 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Created on Fri Jan 14 16:29:08 2022
+
+@author: xp
+"""
 
 import streamlit as st
 
@@ -16,9 +23,11 @@ text = st.text_input("Enter a text for auto completion...", value='Where is Bill
 
 model = st.selectbox("choose a model", ["roberta-base", "bert-base-uncased", "gpt2", "t5"])
 
+data_load_state = st.text('Loading model...')
+nlp = get_model(model)
+
 if text:
-    data_load_state = st.text('Loading model...')
-    nlp = get_model(model)
+
     data_load_state = st.text('Inference to model...')
     result = nlp(text+' '+nlp.tokenizer.mask_token)
     data_load_state.text('')