nickmuchi commited on
Commit
1d76e24
1 Parent(s): 54fae71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -263,7 +263,7 @@ def highlight_entities(article_content,summary_output):
263
  print("")
264
  print(summary_output)
265
 
266
- summary_output = summary_output.replace("/n","")
267
 
268
  print("")
269
  print(summary_output)
@@ -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-cnn_dailymail'
323
  summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
324
  device=0 if torch.cuda.is_available() else -1)
325
  return summarizer
 
263
  print("")
264
  print(summary_output)
265
 
266
+ summary_output = summary_output.replace("n<","")
267
 
268
  print("")
269
  print(summary_output)
 
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