Update app.py

app.py

@@ -1,23 +1,19 @@
 import streamlit as st
 from transformers import pipeline
 
-# Create a T5 text2text-generation pipeline
-pipe = pipeline("text2text-generation", model="kaist-ai/prometheus-13b-v1.0")
+# Create a text2text-generation pipeline with the "google/flan-t5-base" model
+pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 
-# Define the Streamlit app
-st.title("Text Generation App")
+st.title("Text Explanation Model")
+user_text = st.text_area("Enter text:")
 
-# Ask the user for input text
-user_input = st.text_area("Enter text:")
-
-# Generate text when the user clicks the button
-if st.button("Generate Text"):
-    if user_input:
-        # Use the specified pipeline to generate text
-        generated_text = pipe(user_input, max_length=100, do_sample=True)[0]["generated_text"]
-        st.write("Generated Text:")
-        st.write(generated_text)
+if st.button("Generate Explanation"):
+    if user_text:
+        # Use the pipeline to generate an explanation for the input text
+        explanation = pipe(user_text, max_length=100, do_sample=True)[0]["generated_text"]
+        st.markdown("**Explanation:**")
+        st.markdown(explanation)
     else:
         st.warning("Please enter text.")
 
-# Customize the Streamlit app's appearance and other features as needed.
+st.markdown("Use the 'Generate Explanation' button to generate an explanation for the provided text.")