Update app.py

app.py

@@ -4,8 +4,12 @@ import gradio as gr
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 
-text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
-
+# text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
+text_summary = pipeline(
+    "text2text-generation",
+    model="sshleifer/distilbart-cnn-12-6",
+    torch_dtype=torch.float32,   # safer on CPU
+)
 # model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots"
 #               "/a4f8f3ea906ed274767e9906dbaede7531d660ff")
 # text_summary = pipeline("summarization", model=model_path,
@@ -21,9 +25,9 @@ text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
 # Forbes estimates his net worth to be $178 billion.[4]'''
 # print(text_summary(text));
 
-def summary (input):
-    output = text_summary(input)
-    return output[0]['summary_text']
+def summary(input_text):
+    out = text_summary(input_text)
+    return out[0]["generated_text"]
 
 gr.close_all()