import streamlit as st
from transformers import pipeline

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...')
value = st.slider("How long of a response do you want.  The longer the much more time it takes", 
                  min_value=69, max_value=4444, value=100)

if text:
    generated_response_object = pipe(text, max_length=600)
    actual_response = generated_response_object[0]['generated_text']
    st.text(actual_response)