Duplicate from Modfiededition/Writing_Assistant

Co-authored-by: Ashish Goyal <[email protected]>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,38 @@
+---
+title: Writing_Assistant
+emoji: 📈
+colorFrom: yellow
+colorTo: purple
+sdk: streamlit
+app_file: app.py
+pinned: false
+duplicated_from: Modfiededition/Writing_Assistant
+---
+
+# Configuration
+
+`title`: _string_  
+Display title for the Space
+
+`emoji`: _string_  
+Space emoji (emoji-only character allowed)
+
+`colorFrom`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`colorTo`: _string_  
+Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
+
+`sdk`: _string_  
+Can be either `gradio` or `streamlit`
+
+`sdk_version` : _string_  
+Only applicable for `streamlit` SDK.  
+See [doc](https://hf.co/docs/hub/spaces) for more info on supported versions.
+
+`app_file`: _string_  
+Path to your main application file (which contains either `gradio` or `streamlit` Python code).  
+Path is relative to the root of the repository.
+
+`pinned`: _boolean_  
+Whether the Space stays on top of your list.

app.py

@@ -0,0 +1,48 @@
+import streamlit as st
+import transformers
+import tensorflow
+import PIL
+from PIL import Image
+import time
+from transformers import pipeline
+
+model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"
+
+
+@st.cache(allow_output_mutation=True, suppress_st_warning=True)
+def load_model():
+    return pipeline("text2text-generation", model=model_checkpoint)
+model = load_model()
+
+    
+#prompts
+st.title("Writing Assistant for you 🤖")
+st.markdown("This writing assistant detects and corrects grammatical mistakes for you! This assitant uses **T5-base model ✍️** fine-tuned on jfleg dataset.")
+#image = Image.open('new_grammar.jpg')
+#st.image(image, caption='Image Credit: https://abrc.org.au/wp-content/uploads/2020/12/Grammar-checker.jpg')
+st.subheader("Some examples: ")
+example_1 = st.button("I am write on AI")
+example_2 = st.button("This sentence has, bads grammar mistake!")
+
+textbox = st.text_area('Write your text in this box:', '',height=100,  max_chars=500 )
+
+button = st.button('Detect grammar mistakes:')
+
+# output
+st.subheader("Correct sentence: ")
+if example_1:
+    with st.spinner('In progress.......'):
+        output_text = model("I am write on AI")[0]["generated_text"]
+    st.markdown("## "+output_text)
+
+if example_2:
+    with st.spinner('In progress.......'):
+        output_text = model("This sentence has, bads grammar mistake!")[0]["generated_text"]
+    st.markdown("## "+output_text)
+if button:
+    with st.spinner('In progress.......'):
+        if textbox:
+            output_text = model(textbox)[0]["generated_text"]
+        else:
+            output_text = " "
+    st.markdown("## "+output_text)

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+streamlit
+tensorflow
+Pillow