Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the summarization model
+# Load the summarization model (facebook/bart-large-cnn)
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Use the grammar correction model via pipeline
-grammar_correction_pipe = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-synthesis")
+# Use a smaller grammar correction model
+grammar_correction_pipe = pipeline("text2text-generation", model="pszemraj/grammar-synthesis-small")
 
 # Function for grammar correction
 def correct_grammar(user_input):