Spaces:
Sleeping
Sleeping
import streamlit as st | |
import os | |
HF_TOKEN = os.getenv('HF_TOKEN') | |
from huggingface_hub import HfFolder | |
# Set the token using HfFolder (this persists the token) | |
HfFolder.save_token(HF_TOKEN) | |
import transformers | |
from transformers import pipeline | |
generator = pipeline(model="openai-community/gpt2") | |
text = st.text_area("your input") | |
if text: | |
out = generator(text, do_sample=False) | |
st.json(out) | |