File size: 291 Bytes
1282247
 
79173b6
 
7fb6d10
79173b6
 
26f1247
a21cc4c
1
2
3
4
5
6
7
8
9
10
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)