Update app.py

app.py

@@ -42,12 +42,12 @@ class TransformerChatbot(Model):
 st.title("UniGLM TEXT completion Model")
 st.subheader("Next Word Prediction AI Model by Webraft-AI")
 #Picking what NLP task you want to do
-option = st.selectbox('Model',('12M Param')) #option is stored in this variable
+option = st.selectbox('Model',('1')) #option is stored in this variable
 #Textbox for text user is entering
 st.subheader("Enter the text you'd like to analyze.")
 text = st.text_input('Enter word: ') #text is stored in this variable
 
-if option == '12M Param':
+if option == '1':
     loaded_dict = np.load("dict_predict3.bin.npz", allow_pickle=True)
     word_to_num = loaded_dict["word_to_num"].item()
     num_to_word = loaded_dict["num_to_word"].item()