Xhaheen's picture
Update app.py
3f3479d
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 using facebook_OPT_Language_model')
if prompt:
out = generator(prompt)
st.json(out)