Update pages/GPT.py

pages/GPT.py

@@ -4,6 +4,8 @@ import numpy as np
 import transformers 
 import random
 import textwrap
+import warnings
+warnings.filterwarnings('ignore')
 
 @st.cache
 def load_model():
@@ -45,11 +47,8 @@ max_len = st.slider('Length of sequence', 0, 500, 250)
 temp = st.slider('Temperature', 1, 30, 1)
 num_generate = st.text_input("Enter number of sequences")
     
-if st.button('Generate'):# and num_generate and text_input:
-    st.write('Hello')
+if st.button('Generate') and num_generate and text_input:
     prompt = preprocess_text(text_input, tokenizer)
-    st.write('Hello')
     result = predict_sentiment(model, prompt, temp, int(num_generate))
-    st.write('Hello')
     for i in result:
         st.write(textwrap.fill(tokenizer.decode(i), max_len))