alirani commited on
Commit
6d3bd4e
1 Parent(s): b7137a2

fix: add suffix to input

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ button_synopsis = st.button('Get synopsis')
27
 
28
  if button_synopsis:
29
  if len(prod_title.split(' ')) > 0:
30
- gen_synopsis = generate_synopsis(model, tokenizer, prod_title)
31
  st.text_area(gen_synopsis, disabled=True)
32
  else:
33
  st.write('Write a title for the generator to work !')
 
27
 
28
  if button_synopsis:
29
  if len(prod_title.split(' ')) > 0:
30
+ gen_synopsis = generate_synopsis(model, tokenizer, f"{prod_title} : ")
31
  st.text_area(gen_synopsis, disabled=True)
32
  else:
33
  st.write('Write a title for the generator to work !')