Update app.py

app.py

@@ -2,8 +2,20 @@ import streamlit as st
 from transformers import pipeline
 
 pipe = pipeline("sentiment-analysis")
+st.title("Sentiment")
 text = st.text_area("Enter some text here")
 
 if text:
     out = pipe(text)
-    st.json(out)
+    st.json(out)
+
+
+text_generator = pipeline("text-generation")
+st.title("Text Generator")
+input_text = st.text_area("Enter generator prompt: ")
+
+if input_text:
+    st.json(text_generator(input_text))
+
+
+