Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 
 # Summarization
 def summarization(text):
-    image_to_text_model = pipeline("text-generation", model="ainize/bart-base-cnn")
-    summary = image_to_text_model(text, max_length=100, do_sample=False)[0]["generated_text"]
+    text_model = pipeline("text-generation", model="ainize/bart-base-cnn")
+    summary = text_model(text, max_length=100, do_sample=False)[0]["generated_text"]
     return summary
 
 # Sentiment Classification
@@ -14,18 +14,14 @@ def sentiment_classification(summary):
     return result
 
 def main():
-    st.set_page_config(page_title="Your Image to Text Analysis", page_icon="🦜")
+    st.set_page_config(page_title="Your Text Analysis", page_icon="🦜")
     st.header("Tell me your comments!")
-    uploaded_file = st.file_uploader("Select an Image...")
-
-    if uploaded_file is not None:
-        with open(uploaded_file.name, "wb") as file:
-            file.write(uploaded_file.getbuffer())
-        st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    text_input = st.text_input("Enter your text here:")
 
+    if text_input:
         # Stage 1: Summarization
-        st.text('Processing image to text...')
-        summary = summarization(uploaded_file.name)
+        st.text('Processing text...')
+        summary = summarization(text_input)
         st.write(summary)
 
         # Stage 2: Sentiment Classification