nickmuchi commited on
Commit
5e7bda8
1 Parent(s): 1d76e24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -319,7 +319,7 @@ def schleifer_model():
319
 
320
  @st.experimental_singleton(suppress_st_warning=True)
321
  def google_model():
322
- model_name = 'google/pegasus-xsum'
323
  summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
324
  device=0 if torch.cuda.is_available() else -1)
325
  return summarizer
 
319
 
320
  @st.experimental_singleton(suppress_st_warning=True)
321
  def google_model():
322
+ model_name = 'google/pegasus-large'
323
  summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
324
  device=0 if torch.cuda.is_available() else -1)
325
  return summarizer