Update app.py

app.py

@@ -26,6 +26,7 @@ generated_sequences = []
 
 if st.button("Поехали", help="Может занять какое-то время"):
     set_seed()
+    st.title("Генерируем")
     prompt_text = f"{context}"
     encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
     output_sequences = model.generate(
@@ -46,13 +47,15 @@ if st.button("Поехали", help="Может занять какое-то в
         generated_sequence = generated_sequence.tolist()
 
         text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
+        st.write(text)
     
-        text = text[: text.find("</s>") if "</s>" else None]
+       # text = text[: text.find("</s>") if "</s>" else None]
 
-        total_sequence = (
-            prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
-        )
+        #total_sequence = (
+        #    prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
+        #)
     
         # generated_sequences.append(total_sequence)
         # os.system('clear')
-        st.write(total_sequence)
+        st.title("")
+        #st.write(total_sequence)