alfredodeza commited on
Commit
2d8572c
·
1 Parent(s): 40f4abd

make t5 smaller

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", truncation=True, framework="tf")
5
 
6
  def translate(text):
7
  text = text.replace('"', '"')
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-small", truncation=True, framework="tf")
5
 
6
  def translate(text):
7
  text = text.replace('"', '"')