Spaces:

achimoraites
/

Page-Summary

Runtime error

achimoraites commited on Feb 20, 2023

Commit

481c13a

•

1 Parent(s): 2dbf8a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from newspaper import Config, Article
 import torch
 from transformers import pipeline
-summarizer = pipeline("summarization", model="achimoraites/flan-t5-base-xsum")
 def summarize(text):
     return summarizer(text, max_length=128, min_length=30)[0]['summary_text']

 import torch
 from transformers import pipeline
+summarizer = pipeline("summarization", model="achimoraites/flan-t5-base-samsum")
 def summarize(text):
     return summarizer(text, max_length=128, min_length=30)[0]['summary_text']