Update pages/GPT.py

pages/GPT.py

@@ -21,7 +21,7 @@ def preprocess_text(text_input, tokenizer):
 def predict_sentiment(model, prompt, temp, num_generate):
     result = model.generate(
         input_ids=prompt,
-        max_length=150,
+        max_length=100,
         num_beams=5,
         do_sample=True,
         temperature=float(temp),
@@ -47,14 +47,6 @@ else:
 
 if text_input and num_generate:
     prompt = preprocess_text(text_input, tokenizer)
-    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'):
-            try:
-                continue
-            except:
-                st.write('All sequences are return. Generate new')
-        else:
-            None
+        st.write(textwrap.fill(tokenizer.decode(i), max_len))