File size: 643 Bytes
b524623
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
import gradio as gr
# Creating the Q&A pipeline
nlp = pipeline('question-answering', model='deepset/roberta-base-squad2', tokenizer='deepset/roberta-base-squad2')

def questionAndAnswer(ques,content):
  question_set = {'question':ques,'context':content}
  results = nlp(question_set)
  return results['answer']

interface = gr.Interface(fn=questionAndAnswer, 
                        inputs=["text","text"],
                         outputs="text", 
                        title='Bilal Question&Answer')
                        

interface.launch(inline=False)