Update pages/GPT.py

pages/GPT.py

@@ -3,6 +3,7 @@ import torch
 import numpy as np
 import transformers 
 import random
+import textwrap
 
 def load_model():
     model_finetuned = transformers.AutoModelWithLMHead.from_pretrained(
@@ -44,9 +45,10 @@ if st.button('Generate a random number of sequences'):
 else:
     num_generate = st.text_input("Enter number of sequences")
 
-if text_input:
+if text_input and num_generate:
     prompt = preprocess_text(text_input, tokenizer)
-    result = predict_sentiment(model, prompt, temp, num_generate)
+    print(prompt, temp, num_generate)
+    result = predict_sentiment(model, prompt, temp, int(num_generate))
     for i in result:
         st.write(textwrap.fill(tokenizer.decode(i), max_len))
         if st.button('Next'):
@@ -55,4 +57,4 @@ if text_input:
             except:
                 st.write('All sequences are return. Generate new')
         else:
-            None
+            None