Update app.py

app.py

@@ -22,5 +22,7 @@ text_generator = load_text_gen_model()
 action = st.sidebar.selectbox("Pick an Action", ["Generate an Article","Create an Image"])
 
 if action == "Generate an Article":
-    st.text_input("Enter a prompt")
+    prompt = st.text_input("Enter a prompt")
+    res = generator( prompt, max_length=100, temperature=0.7)
+    st.write(res)