Spaces:

derina
/

BartSummarizer

Runtime error

derina commited on Sep 25, 2023

Commit

3387cd0

1 Parent(s): ae6f6f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-summarizer = pipeline("summarization")
 def summarize(text):
   return summarizer(text, max_length=400, min_length=50)[0]['summary_text']

 import gradio as gr
 from transformers import pipeline
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 def summarize(text):
   return summarizer(text, max_length=400, min_length=50)[0]['summary_text']