File size: 744 Bytes
86f1f69
 
 
 
 
 
 
 
 
 
 
 
 
c279e3a
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
import gradio as grad
import ast

_model = "deepset/roberta-base-squad2"
_pipeline = pipeline("question-answering", model = _model, tokenizer = _model)

def answer_question(question, context):
    text = "{" + 'question': '"+question+"', 'context':'"+context+"'}"
    di = ast.literal_eval(text)
    response = _pipeline(di)
    return response

grad.Interface(answer_question, inputs=["text", "text"], outputs="text").launch()

text = grad.inputs.Textbox(placeholder="Lets chat together")
grad.Interface (fn=converse,
               theme="default",
               inputs=[text, "state"],
               outputs=["html", "state"],
               css=css).launch()