File size: 359 Bytes
e94484a
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import gradio as gr
from transformers import pipeline

def chatbot(input_text):
    generator = pipeline('text-generation', model='gpt2')
    response = generator(input_text, max_length=50, num_return_sequences=1)
    return response[0]['generated_text']

iface = gr.Interface(fn=chatbot, inputs="text", outputs="text", title="Gradio Chatbot")
iface.launch()