Update app.py

app.py

@@ -1,19 +1,12 @@
-from transformers import pipeline
-unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
-unmasker("Hello I'm a [MASK] model.")
-
-
 import streamlit as st
 from transformers import pipeline
 
 def summarize_text(text):
     unmasker = pipeline('fill-mask', model='distilbert-base-uncased')
-    unmasker = summarizer(text,do_sample=True)[0]['summary_text']
-    return unmasker
+    text_input = unmasker(text, do_sample=True)[0]['text_input']
+    return text_input
 
 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)