Spaces:
Sleeping
Sleeping
Hugo
commited on
Commit
·
96b13ff
1
Parent(s):
671d64d
update model v2 (w gen headlines)
Browse files
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:")
|