text-generation / app.py
mwaseemrandhawa's picture
Update app.py
26f1247
import streamlit as st
from transformers import pipeline
from transformers import pipeline
pipe = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
text = st.text_area("Enter some text")
if text:
out = pipe(text, max_length=50, do_sample=True, temperature=0.9)
st.write(out)