Update pages/GPT.py

pages/GPT.py

@@ -5,7 +5,7 @@ import transformers
 import random
 import textwrap
 
-@st.cache
+@st.cache_data
 def load_model():
     model_finetuned = transformers.AutoModelWithLMHead.from_pretrained(
         'tinkoff-ai/ruDialoGPT-small',
@@ -46,13 +46,14 @@ max_len = st.slider('Length of sequence', 0, 500, 250)
 print(max_len)
 temp = st.slider('Temperature', 1, 30, 1)
 print(temp)
+
 if st.button('Generate a random number of sequences'):
     num_generate = random.randint(1,5)
     st.write(f'Number of sequences: {num_generate}')
 else:
     num_generate = st.text_input("Enter number of sequences")
 
-if text_input and num_generate:
+if st.button('Generate') and num_generate and text_input:
     prompt = preprocess_text(text_input, tokenizer)
     result = predict_sentiment(model, prompt, temp, int(num_generate))
     for i in result: