Update app.py

app.py

@@ -1,19 +1,5 @@
-import streamlit as st
 from transformers import pipeline
 
-def summarize_text(text):
-    summarizer = pipeline("summarization", model="casarf/comment_model_test")
-    summary_text = summarizer(text,do_sample=True)[0]['summary_text']
-    return summary_text
+unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
 
-def main():
-    st.title("Text Summarization App")
-    st.write("Enter your text below:")
-    user_input = st.text_area("Input Text Here")
-    if st.button("Summarize"):
-        summary = summarize_text(user_input)
-        st.write("Summary:")
-        st.write(summary)
-
-if __name__ == "__main__":
-    main()
+unmasker("Hello I'm a [MASK] model.")