nickmuchi commited on
Commit
c250e24
1 Parent(s): 12da133

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -341,7 +341,7 @@ def schleifer_model():
341
  @st.experimental_singleton(suppress_st_warning=True)
342
  def google_model():
343
  model_name = 'google/pegasus-cnn_dailymail'
344
- summarizer = pipeline('summarization',model=model=model_name, tokenizer=model_name,
345
  device=0 if torch.cuda.is_available() else -1)
346
  return summarizer
347
 
 
341
  @st.experimental_singleton(suppress_st_warning=True)
342
  def google_model():
343
  model_name = 'google/pegasus-cnn_dailymail'
344
+ summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
345
  device=0 if torch.cuda.is_available() else -1)
346
  return summarizer
347