Hugo commited on
Commit
96b13ff
·
1 Parent(s): 671d64d

update model v2 (w gen headlines)

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -85,7 +85,8 @@ if st.button("Generate Headline"):
85
  if content.strip():
86
  if not video_summary.strip():
87
  video_summary = ''
88
- prompt = process_prompt(tokenizer, content, video_summary, guidelines)
 
89
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
90
 
91
  st.write("### Generated 5 Potential Headlines:")
 
85
  if content.strip():
86
  if not video_summary.strip():
87
  video_summary = ''
88
+ # prompt = process_prompt(tokenizer, content, video_summary, guidelines)
89
+ prompt = process_prompt(tokenizer, content, video_summary)
90
  inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
91
 
92
  st.write("### Generated 5 Potential Headlines:")