Hugging Face's logo

Spaces:
EE21
/
ToS-Summarization
Runtime error

App Files Community
EE21 commited on
Commit
19e7d09
•
1 Parent(s): bf2b9e4

Update abstractive_model.py

Browse files
Files changed (1) hide show
  1. abstractive_model.py +1 -1
abstractive_model.py CHANGED
@@ -1,4 +1,4 @@
1
- from transformers import pipeline
2
 
3
  # Load the BART tokenizer and model
4
  tokenizer = AutoTokenizer.from_pretrained("EE21/BART-ToSSimplify")
 
1
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
 
3
  # Load the BART tokenizer and model
4
  tokenizer = AutoTokenizer.from_pretrained("EE21/BART-ToSSimplify")