Create app.py

app.py

@@ -0,0 +1,40 @@
+import torch
+import gradio as gr
+from transformers import pipeline,BartTokenizer
+
+
+#custom_cache_dir = '/Users/walidahmed/huggingface_models'
+#tokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-cnn-12-6', cache_dir=custom_cache_dir)
+
+
+
+model_path="../models/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
+text_summary = pipeline("summarization", model=model_path, torch_dtype=torch.bfloat16)
+
+
+text='''Elon Reeve Musk (/ˈiːlɒn/ EE-lon; born June 28, 1971) is a businessman and investor.
+ He is the founder, chairman, CEO, and CTO of SpaceX; angel investor, CEO, product architect,
+ and former chairman of Tesla, Inc.; owner, executive chairman, and CTO of X Corp.;
+ founder of the Boring Company and xAI; co-founder of Neuralink and OpenAI; and president
+ of the Musk Foundation. He is one of the wealthiest people in the world; as of April 2024,
+ Forbes estimates his net worth to be $178 billion.[4]'''
+
+# print(text_summary(text));
+# print(text_summary(text));
+
+
+def summary (input):
+    output = text_summary(input)
+    return output[0]['summary_text']
+
+print(summary (text))
+
+gr.close_all()
+
+# demo = gr.Interface(fn=summary, inputs="text",outputs="text")
+demo = gr.Interface(fn=summary,
+                    inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
+                    outputs=[gr.Textbox(label="Summarized text",lines=4)],
+                    title=" Project 1: Text Summarizer",
+                    description="Summarize the text")
+demo.launch()