| 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() |
|
|
