import gradio as gr
from transformers import pipeline


question_answerer = pipeline("question-answering", model="shashankheg/my_qa_model")


def QA(Context,Question):
    result=question_answerer(context=Context,question=Question)
    for answer in result:
        return result['answer']

    
interface = gr.Interface(fn =QA,
                         inputs = ['text','text'],
                         outputs=['text'],
                         title= 'Qustion Answering using Distilbert',
                         theme = gr.themes.Glass(),
                         examples = [
                            ['On February 6, 2016, one day before her performance at the Super Bowl, Beyoncé released a new single exclusively on music streaming service Tidal called "Formation"','When did Beyoncé release Formation?'],
                            ['Special bundles of the game contain a Wolf Link Amiibo figurine, which unlocks a Wii U-exclusive dungeon called the "Cave of Shadows" and can carry data over to the upcoming 2016 Zelda game. Other Zelda-related Amiibo figurines have distinct functions: Link and Toon Link replenish arrows, Zelda and Sheik restore Link\'s health, and Ganondorf causes Link to take twice as much damage.','What characters will be able to replenish arrows?'],

                         ],                 
                         
                         css = """"
                         body {background-color: #FFCCCB}"""
                         
                         )
                         
                        
interface.launch()