Update app.py

app.py

@@ -62,11 +62,8 @@ def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
         total_sequence = (
             text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
         )
-        print(total_sequence)
-        
-        #generated_sequences.append(total_sequence)
-        generated_sequences.append(text)
-        st.write(text)
+        generated_sequences.append(total_sequence)
+        st.write(total_sequence)
         
     parsed_text = total_sequence.replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n")
     if len(parsed_text) == 0: