File size: 388 Bytes
e9d63b2 6b7e9c6 e9d63b2 6b7e9c6 d6e07a2 6b7e9c6 d6e07a2 6b7e9c6 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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) |