Leo9Bot / app.py
ehi20011's picture
Update app.py
66557b5 verified
Raw
History Blame
433 Bytes
import gradio as gr
from transformers import pipeline
qa = pipeline("text2text-generation", model="google/flan-t5-base")
def answer_question(prompt):
output = qa(prompt, max_length=256, do_sample=False)
return output[0]['generated_text']
gr.Interface(fn=answer_question,
inputs=gr.Textbox(label="Ask EduBot"),
outputs=gr.Textbox(label="Answer"),
title="Leo9 EduBot (CPU)").launch()