Create app.py

app.py

@@ -0,0 +1,49 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+
+# Load models
+grammar_model_name = "prithivida/grammar_error_correcter_v1"
+summarizer_model_name = "facebook/bart-large-cnn"
+title_model_name = "t5-base"
+
+# Load tokenizer and models
+grammar_tokenizer = AutoTokenizer.from_pretrained(grammar_model_name)
+grammar_model = AutoModelForSeq2SeqLM.from_pretrained(grammar_model_name)
+
+title_tokenizer = AutoTokenizer.from_pretrained(title_model_name)
+title_model = AutoModelForSeq2SeqLM.from_pretrained(title_model_name)
+
+summarizer = pipeline("summarization", model=summarizer_model_name)
+
+# Define functions
+def polish_abstract(text):
+    # Grammar correction
+    inputs = grammar_tokenizer.encode("gec: " + text, return_tensors="pt", max_length=512, truncation=True)
+    outputs = grammar_model.generate(inputs, max_length=512, num_beams=5, early_stopping=True)
+    corrected = grammar_tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    # Summarization
+    summary = summarizer(corrected, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
+
+    # Title generation
+    title_input = "generate title: " + corrected
+    title_inputs = title_tokenizer.encode(title_input, return_tensors="pt", max_length=512, truncation=True)
+    title_outputs = title_model.generate(title_inputs, max_length=15, num_beams=5, early_stopping=True)
+    title = title_tokenizer.decode(title_outputs[0], skip_special_tokens=True)
+
+    return corrected, summary, title
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=polish_abstract,
+    inputs=gr.Textbox(lines=10, label="Paste Your Raw Abstract Here"),
+    outputs=[
+        gr.Textbox(label="✅ Corrected Abstract (Grammar Fixed)"),
+        gr.Textbox(label="🪄 Polished Summary (Concise Version)"),
+        gr.Textbox(label="📘 Suggested Title"),
+    ],
+    title="🧠 AI Abstract & Title Polisher",
+    description="Paste your raw abstract. The AI will fix grammar, generate a concise summary, and suggest a title."
+)
+
+iface.launch()