Gladiator commited on
Commit
ae25549
1 Parent(s): d6f77b5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def abstractive_summarizer(text : str, model):
28
 
29
  @st.cache()
30
  def load_abs_model():
31
- model = T5ForConditionalGeneration.from_pretrained('t5-large')
32
  return model
33
 
34
 
 
28
 
29
  @st.cache()
30
  def load_abs_model():
31
+ model = T5ForConditionalGeneration.from_pretrained('t5-base')
32
  return model
33
 
34