Update app.py

app.py

@@ -1,17 +1,20 @@
-# First, ensure the 'transformers' library is installed
-try:
-    from transformers import pipeline
-except ModuleNotFoundError:
-    import subprocess
-    subprocess.check_call(["pip", "install", "transformers"])
-    from transformers import pipeline
+from transformers import pipeline
+import streamlit as st
 
-# Example: Load a grammar correction pipeline (you can replace this with your preferred task/model)
-corrector = pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
+# Load grammar correction pipeline
+@st.cache_resource
+def load_model():
+    return pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
 
-# Sample sentence
-sentence = "She don't like going to the gym because it make her tired."
+corrector = load_model()
 
-# Run correction
-result = corrector(sentence, max_length=100, clean_up_tokenization_spaces=True)
-print("Corrected Sentence:", result[0]['generated_text'])
+# Streamlit UI
+st.title("Grammar Correction Assistant")
+
+user_input = st.text_area("Enter a sentence to correct:", "She don't like going to the gym because it make her tired.")
+
+if st.button("Correct Sentence"):
+    with st.spinner("Correcting..."):
+        result = corrector(user_input, max_length=100, clean_up_tokenization_spaces=True)
+        corrected_sentence = result[0]['generated_text']
+        st.markdown(f"**Corrected Sentence:** {corrected_sentence}")