Update app.py

app.py

@@ -3,11 +3,12 @@ import gradio as gr
 from transformers import pipeline
 
 # Create a summarization pipeline
-summarizer = pipeline("summarization")
+# summarizer = pipeline("summarization")
+text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
 
 # Define a function that takes a text and returns a summary
 def summarize(text):
-  summary = summarizer(text, max_length=150, min_length=40, do_sample=False)[0]
+  summary = text_summery(text, max_length=150, min_length=40, do_sample=False)[0]
   return summary["summary_text"]
 
 # Create a Gradio interface