DavidWill commited on
Commit
2592248
·
verified ·
1 Parent(s): 019df66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,11 +4,12 @@ import gradio as gr
4
  import trafilatura
5
 
6
  # 使用公開可用的多語摘要模型
 
7
  summarizer = pipeline(
8
  "summarization",
9
- model="pszemraj/long-t5-tglobal-base-16384-book-summary",
10
- tokenizer="pszemraj/long-t5-tglobal-base-16384-book-summary",
11
- device=-1 # CPU
12
  )
13
 
14
  def summarize_url(url):
 
4
  import trafilatura
5
 
6
  # 使用公開可用的多語摘要模型
7
+ #from transformers import pipeline
8
  summarizer = pipeline(
9
  "summarization",
10
+ model="csebuetnlp/mT5_multilingual_XLSum",
11
+ tokenizer="csebuetnlp/mT5_multilingual_XLSum",
12
+ device=-1
13
  )
14
 
15
  def summarize_url(url):