| 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) |