Update app.py

app.py

@@ -1,11 +1,14 @@
+# Install required packages if not already installed
+!pip install streamlit transformers
+
 import streamlit as st
 from transformers import pipeline
 
-# Load the summarization model (facebook/bart-large-cnn)
+# Load the summarization model
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Use a smaller grammar correction model
-grammar_correction_pipe = pipeline("text2text-generation", model="pszemraj/grammar-synthesis-small")
+# Load the grammar correction model
+grammar_correction_pipe = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-synthesis")
 
 # Function for grammar correction
 def correct_grammar(user_input):
@@ -38,12 +41,12 @@ task = st.selectbox("Choose a task", ["Summarize Text", "Correct Grammar"])
 # Input component for text
 user_input = st.text_area("Enter your text here:")
 
-# Button to submit the input
+# Process and display output based on selected task
 if st.button("Submit"):
     if task == "Summarize Text":
         output = correct_and_summarize(user_input)  # Correct grammar, then summarize
     elif task == "Correct Grammar":
         output = correct_grammar(user_input)  # Only correct grammar
     
-    # Display the result
-    st.text_area("Output", output, height=200)
+    # Display the output
+    st.text_area("Output", output, height=200, disabled=True)