eskayML's picture
Create app.py
6e8a47e
raw
history blame
438 Bytes
from transformers import pipeline
import gradio as gr
from transformers.pipelines.conversational import Conversation
def ama_func(x):
conversation = Conversation(x)
return conversational_pipeline(conversation)
demo = gr.Interface(
fn = ama_func,
inputs = gr.Textbox(label="Hey, Bakayaro I'm the salty conversation bot 🤨", placeholder = 'Enter your question here....', lines = 3),
outputs = 'text',
)
demo.launch()