Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
# Create a text generation pipeline using Hugging Face Transformers | |
generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B") | |
# Define the chat function | |
def chat(input_text): | |
response = generator(input_text, max_length=50, do_sample=True)[0]["generated_text"] | |
return response | |
# Create a Gradio interface | |
iface = gr.Interface( | |
fn=chat, | |
inputs=gr.inputs.Textbox(label="Input Text"), | |
outputs=gr.outputs.Textbox(label="Response"), | |
live=True, | |
) | |
# Launch the Gradio interface | |
iface.launch() | |