sanshizhang commited on
Commit
7442775
β€’
1 Parent(s): 43bb98f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
 
4
  # εŠ θ½½δΈ­ζ–‡ζ‘˜θ¦ζ¨‘εž‹
5
- summarizer_tokenizer = AutoTokenizer.from_pretrained("IDEA-CCNL/Randeng-T5-77M-Summary-Chinese")
6
- summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("IDEA-CCNL/Randeng-T5-77M-Summary-Chinese")
7
  summarizer = pipeline("summarization", model=summarizer_model, tokenizer=summarizer_tokenizer)
8
 
9
  # εŠ θ½½δΈ­ζ–‡ζƒ…ζ„Ÿεˆ†ζžζ¨‘εž‹
 
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
3
 
4
  # εŠ θ½½δΈ­ζ–‡ζ‘˜θ¦ζ¨‘εž‹
5
+ summarizer_tokenizer = AutoTokenizer.from_pretrained("fnlp/bart-base-chinese")
6
+ summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("fnlp/bart-base-chinese")
7
  summarizer = pipeline("summarization", model=summarizer_model, tokenizer=summarizer_tokenizer)
8
 
9
  # εŠ θ½½δΈ­ζ–‡ζƒ…ζ„Ÿεˆ†ζžζ¨‘εž‹