Adding model to app

app.py

@@ -1,5 +1,13 @@
 import streamlit as st
+from transformers import pipeline
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+model_location = "KingZack/future-futurama-maker"
+pipe = pipeline("text-generation", model=model_location)
+
+text = st.text_area('Give me ideas to generate a new Futurama episode...')
+
+if text:
+    generated_response_object = pipe(text, max_length=333)
+    actual_response = generated_response_object[0]['generated_text']
+    st.text(actual_response)