Spaces:
Runtime error
Runtime error
#for text generation | |
import gradio as gr | |
from transformers import pipeline | |
text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B") | |
def generate_text(prompt): | |
generated_text = text_generator(prompt, max_length=200)[0]["generated_text"] | |
return generated_text | |
iface1 = gr.Interface( | |
generate_text, | |
inputs="text", | |
outputs="text", | |
title="Hugging Face Text Generation", | |
description="Generate text using Hugging Face's GPT-Neo 2.7B model.", | |
examples=[ | |
["The quick brown fox",], | |
["Once upon a time",], | |
["In a galaxy far, far away",], | |
], | |
) | |
iface1.launch() |