Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ if st.button("Summarize"):
|
|
40 |
st.text("It may take a minute or two.")
|
41 |
nwords=len(input_text.split(" "))
|
42 |
text_input_ids=tokenizer('summarize: '+input_text, max_length=600, padding="max_length", truncation=True).input_ids
|
43 |
-
output_ids = model.generate(text_input_ids)
|
44 |
generated_summary = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
45 |
|
46 |
st.header("Summary")
|
|
|
40 |
st.text("It may take a minute or two.")
|
41 |
nwords=len(input_text.split(" "))
|
42 |
text_input_ids=tokenizer('summarize: '+input_text, max_length=600, padding="max_length", truncation=True).input_ids
|
43 |
+
output_ids = model.generate(torch.tensor(text_input_ids))
|
44 |
generated_summary = tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
45 |
|
46 |
st.header("Summary")
|