sanshizhang commited on
Commit
d43b643
1 Parent(s): 7442775

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
 
4
  # 加载中文摘要模型
5
  summarizer_tokenizer = AutoTokenizer.from_pretrained("fnlp/bart-base-chinese")
 
1
  import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForSequenceClassification, pipeline
3
 
4
  # 加载中文摘要模型
5
  summarizer_tokenizer = AutoTokenizer.from_pretrained("fnlp/bart-base-chinese")