import os
import cherrypy
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
model_name = "deepset/roberta-base-squad2"
path = os.path.abspath(os.path.dirname(__file__))
config = {
'global' : {
'server.socket_host' : '127.0.0.1',
'server.socket_port' : 8080,
'server.thread_pool' : 8,
},
}
class Root(object):
def answer_question(self, question, context):
question_answerer = pipeline('question-answering', model=model_name, tokenizer=model_name)
answer = question_answerer({'question': question, 'context': context})
print(answer)
return answer["answer"]
@cherrypy.expose
def index(self, *args, **kwargs):
answer = ""
with open('data.txt') as f:
context = f.read()
f.close()
if cherrypy.request.params.get('question'):
question_answerer = pipeline('question-answering', model=model_name, tokenizer=model_name)
temp = question_answerer({'question': cherrypy.request.params.get('question'), 'context': context})
answer = "answer
{}
".format(temp["answer"])
return "reinout project 1question
{}".format(answer)
@cherrypy.expose
def question(self, *args,**kwargs):
with open('data.txt') as f:
alles = f.read()
f.close()
string=self.answer_question(cherrypy.request.params.get('the_link'), alles)
return string
if __name__ == '__main__':
cherrypy.quickstart(Root(), '/', config)