dokster commited on
Commit
d18e830
·
1 Parent(s): f4fa150

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -56,7 +56,7 @@ def generate(prompt, temperature, max_length, top_k, top_p, num_return_sequences
56
  )
57
 
58
  for i, sample_output in enumerate(sample_outputs):
59
- answers.append("{}".format(tokenizer.decode(sample_output, skip_special_tokens=True)))
60
 
61
  return answers
62
 
@@ -110,9 +110,10 @@ def main():
110
 
111
  a, b = st.columns([4, 1])
112
  user_input = a.text_input(
113
- label="Your message:",
114
  placeholder="Напишите затравку для шутки или скетча...",
115
- label_visibility="collapsed"
 
116
  )
117
  button = b.button("Отправить", use_container_width=True)
118
 
 
56
  )
57
 
58
  for i, sample_output in enumerate(sample_outputs):
59
+ answers.append(tokenizer.decode(sample_output, skip_special_tokens=True).replace('\n', ' '))
60
 
61
  return answers
62
 
 
110
 
111
  a, b = st.columns([4, 1])
112
  user_input = a.text_input(
113
+ label="Напишите затравку для шутки или скетча...",
114
  placeholder="Напишите затравку для шутки или скетча...",
115
+ # label_visibility="collapsed",
116
+ value='Зашла улитка в бар'
117
  )
118
  button = b.button("Отправить", use_container_width=True)
119