import streamlit as st
from transformers import pipeline

generator = pipeline('text-generation', model="facebook/opt-125m")
prompt = st.text_area('Enter text below to generate new text!')

if prompt:
  out = generator(prompt)
  st.json(out)