paragon-analytics commited on
Commit
f6468ff
·
1 Parent(s): ea558ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ tokenizer = AutoTokenizer.from_pretrained("paragon-analytics/bert_resil")
28
  model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/bert_resil")
29
 
30
  para_tokenizer = AutoTokenizer.from_pretrained("paragon-analytics/t5_para")
31
- para_model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/t5_para")
32
 
33
  kw_extractor = yake.KeywordExtractor()
34
  custom_kw_extractor = yake.KeywordExtractor(lan="en", n=2, dedupLim=0.2, top=10, features=None)
 
28
  model = AutoModelForSequenceClassification.from_pretrained("paragon-analytics/bert_resil")
29
 
30
  para_tokenizer = AutoTokenizer.from_pretrained("paragon-analytics/t5_para")
31
+ para_model = AutoModelForSeq2SeqLM.from_pretrained("paragon-analytics/t5_para")
32
 
33
  kw_extractor = yake.KeywordExtractor()
34
  custom_kw_extractor = yake.KeywordExtractor(lan="en", n=2, dedupLim=0.2, top=10, features=None)