Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import pipeline
 
 # Load NLP pipelines
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-paraphraser = pipeline("text2text-generation", model="Vamsi/T5_Paraphrase_Paws")
 en_to_fil_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-tl")
 fil_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-tl-en")
 grammar_corrector = pipeline("text2text-generation", model="prithivida/grammar_error_correcter_v1")
@@ -14,12 +13,9 @@ def process_text(text, task):
 
     try:
         if task == "Summarize":
-            summary = summarizer(text, max_length=100, min_length=30, do_sample=False)
+            summary = summarizer(text, max_length=512, min_length=30, do_sample=False)
             return summary[0]['summary_text']
         
-        elif task == "Paraphrase":
-            return paraphraser(text)[0]['generated_text']
-        
         elif task == "Translate English to Filipino":
             return en_to_fil_translator(text)[0]['translation_text']
         
@@ -27,7 +23,7 @@ def process_text(text, task):
             return fil_to_en_translator(text)[0]['translation_text']
         
         elif task == "Grammar & Spell Check":
-            corrected = grammar_corrector(f"gec: {text}")
+            corrected = grammar_corrector(f"gec: {text}", max_length=512, min_length=30, do_sample=False)
             return corrected[0]['generated_text']
         
         else:
@@ -41,7 +37,6 @@ iface = gr.Interface(
         gr.Textbox(lines=10, placeholder="Paste your text here..."),
         gr.Dropdown([
             "Summarize",
-            "Paraphrase",
             "Translate English to Filipino",
             "Translate Filipino to English",
             "Grammar & Spell Check"
@@ -49,7 +44,7 @@ iface = gr.Interface(
     ],
     outputs="text",
     title="Smart Text Assistant",
-    description="Use NLP to summarize, paraphrase, translate (English ↔ Filipino), and correct grammar and spelling."
+    description="Use NLP to summarize, translate (English ↔ Filipino), and correct grammar and spelling."
 )
 
 if __name__ == "__main__":