Update pages/GPT.py

pages/GPT.py

@@ -5,7 +5,7 @@ import transformers
 import random
 import textwrap
 
-@st.cache_data
+# @st.cache_data
 def load_model():
     model_finetuned = transformers.AutoModelWithLMHead.from_pretrained(
         'tinkoff-ai/ruDialoGPT-small',
@@ -33,7 +33,6 @@ 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')
@@ -41,20 +40,20 @@ st.title('Text generation with dreambook')
 model, tokenizer = load_model()
 
 text_input = st.text_input("Enter some text about movie")
-print(text_input)
 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 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:
         st.write(textwrap.fill(tokenizer.decode(i), max_len))