Update app.py

app.py

@@ -4,40 +4,33 @@ import gradio as gr
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 
-# model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff")
-# text_summary = pipeline("summarization", model=model_path, torch_dtype=torch.bfloat16)
-text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.float16)
-# text_summary = pipeline("text-generation", model="sshleifer/distilbart-cnn-12-6")
-# tokenizer = text_summary.tokenizer
-# text='''Narendra Damodardas Modi[a] (born 17 September 1950) is an Indian politician who has served as the prime minister of India since 2014. Modi was the chief minister of Gujarat from 2001 to 2014 and is the member of parliament (MP) for Varanasi. He is a member of the Bharatiya Janata Party (BJP) and of the Rashtriya Swayamsevak Sangh (RSS), a right-wing Hindutva paramilitary volunteer organisation. He is the longest-serving prime minister outside the Indian National Congress.'''
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
+
+# 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));
 
-def summary(input):  
-    output = text_summary(
-        input, 
-        max_new_tokens=60, 
-        do_sample=False, 
-        pad_token_id=tokenizer.pad_token_id,
-        num_beams=4,           # ← ADD THIS
-        no_repeat_ngram_size=3, # ← ADD THIS  
-        early_stopping=True     # ← ADD THIS
-    )
-    return output[0]['generated_text'].strip()
+def summary (input):
+    output = text_summary(input)
+    return output[0]['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", lines=10, placeholder="Paste your text here..."),
-    outputs=gr.Textbox(label="📄 Summary", lines=6),
-    title="AI Text Summarizer",
-    description="Paste any text and get a concise summary powered by DistilBART-CNN.",
-    cache_examples=False
-
-)
-
-demo.launch(theme=gr.themes.Soft())
-#
-# import gradio as gr
-# print(gr.__file__)  # Should now show site-packages path
-# print(hasattr(gr, "Interface"))  # Should print: True
+
+# 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="@GenAILearniverse Project 1: Text Summarizer",
+                    description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE TEXT")
+demo.launch()