madanagrawal commited on
Commit
30dbffe
·
verified ·
1 Parent(s): 34f6eaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ st.title('Text Summarization')
5
  text = st.text_input
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("madanagrawal/summarization_model")
8
- inputs = tokenizer(text, return_tensors="pt").input_ids
9
 
10
  from transformers import AutoModelForSeq2SeqLM
11
 
 
5
  text = st.text_input
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("madanagrawal/summarization_model")
8
+ inputs = tokenizer(text[0], return_tensors="pt").input_ids
9
 
10
  from transformers import AutoModelForSeq2SeqLM
11