Create app.py

app.py

@@ -0,0 +1,37 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import gradio as gr
+
+# Load model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
+
+def summarize_text(text):
+    # Tokenize the input text
+    inputs = tokenizer([text], max_length=1024, truncation=True, return_tensors="pt")
+    
+    # Generate summary
+    summary_ids = model.generate(
+        inputs["input_ids"], 
+        num_beams=4, 
+        max_length=200, 
+        early_stopping=True
+    )
+    
+    # Decode and return the summary
+    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary
+
+# Create Gradio interface
+interface = gr.Interface(
+    fn=summarize_text,
+    inputs=gr.Textbox(lines=10, placeholder="Paste your text here...", label="Input Text"),
+    outputs=gr.Textbox(lines=5, label="Summary"),
+    title="Text Summarizer with BART-large-CNN",
+    description="Paste any long text and get a concise summary using Facebook's BART-large-CNN model.",
+    examples=[
+        ["The Apollo program, also known as Project Apollo, was the third United States human spaceflight program carried out by the National Aeronautics and Space Administration (NASA), which succeeded in preparing and landing the first humans on the Moon from 1968 to 1972. It was first conceived during Dwight D. Eisenhower's administration as a three-person spacecraft to follow the one-person Project Mercury, which put the first Americans in space. Apollo was later dedicated to President John F. Kennedy's national goal of landing a man on the Moon and returning him safely to the Earth by the end of the 1960s, which he proposed in a May 25, 1961, address to Congress."]
+    ]
+)
+
+# Launch the app
+interface.launch()