Spaces:
Runtime error
Runtime error
File size: 546 Bytes
23d152f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from flask import Blueprint, request
from flask_restful import Api, Resource
from chatBot.common.utils import getAnswerLlama, getAnswerGpt
resources = Blueprint('resources', __name__)
api = Api(resources)
class Gpt(Resource):
def get(self):
question = request.json['question']
return {'answer': getAnswerGpt(question)}
api.add_resource(Gpt, '/gpt')
class Llama(Resource):
def get(self):
question = request.json['question']
return {'answer': getAnswerLlama(question)}
api.add_resource(Llama, '/llama') |