import streamlit as st

from transformers import pipeline

text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=0)  # You can choose a different model if needed

# Generate text
prompt = "Once upon a time"
generated_text = text_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']

# Print the generated text
print(generated_text)