bankholdup commited on
Commit
41e6718
1 Parent(s): 73288d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -28,7 +28,6 @@ context = st.text_input("Введите начало песни", "Как дел
28
  generated_sequences = []
29
 
30
  if st.button("Поехали", help="Может занять какое-то время"):
31
- st.title(f"Новая топовая песня")
32
  prompt_text = f"{context}"
33
  encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
34
  output_sequences = model.generate(
@@ -58,7 +57,7 @@ if st.button("Поехали", help="Может занять какое-то в
58
 
59
  # Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
60
  total_sequence = (
61
- prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
62
  )
63
 
64
  generated_sequences.append(total_sequence)
 
28
  generated_sequences = []
29
 
30
  if st.button("Поехали", help="Может занять какое-то время"):
 
31
  prompt_text = f"{context}"
32
  encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
33
  output_sequences = model.generate(
 
57
 
58
  # Add the prompt at the beginning of the sequence. Remove the excess text that was used for pre-processing
59
  total_sequence = (
60
+ prompt_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :] + "/n"
61
  )
62
 
63
  generated_sequences.append(total_sequence)