ZinebSN commited on
Commit
01b15b8
1 Parent(s): 9735c16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")