Update pages/GPT.py

pages/GPT.py

@@ -5,7 +5,7 @@ import transformers
 import random
 import textwrap
 
-@st.cache_resource
+@st.cache
 def load_model():
     model_finetuned = transformers.AutoModelWithLMHead.from_pretrained(
         'tinkoff-ai/ruDialoGPT-small',
@@ -33,6 +33,7 @@ def predict_sentiment(model, prompt, temp, num_generate):
             no_repeat_ngram_size=3,
             num_return_sequences=num_generate,
             ).cpu().numpy()
+    print(result)
     return result
 
 st.title('Text generation with dreambook')
@@ -44,8 +45,10 @@ 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:
+if st.button('Generate'):# and num_generate and text_input:
+    st.write('Hello')
     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: