Update app.py
Browse files
app.py
CHANGED
@@ -2,11 +2,10 @@ import streamlit as st
|
|
2 |
from transformers import (
|
3 |
AutoModelForSeq2SeqLM,
|
4 |
AutoModelForTokenClassification,
|
5 |
-
AutoTokenizer
|
6 |
-
warnings)
|
7 |
|
8 |
#忽略报错
|
9 |
-
warnings.filterwarnings("ignore")
|
10 |
|
11 |
tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
|
12 |
extractive_summary = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
|
|
2 |
from transformers import (
|
3 |
AutoModelForSeq2SeqLM,
|
4 |
AutoModelForTokenClassification,
|
5 |
+
AutoTokenizer)
|
|
|
6 |
|
7 |
#忽略报错
|
8 |
+
#warnings.filterwarnings("ignore")
|
9 |
|
10 |
tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
|
11 |
extractive_summary = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|