def conditions(api: infermedica_api.API):
Log.info('Conditions list:')
Log.pretty(api.conditions_list())
Log.info('Conditions details:')
Log.pretty(api.condition_details('c_221'))
Log.warn('Conditions details (Fail Test)')
Log.pretty(api.condition_details('fail'))
})
api = infermedica_api.get_api()
# resp = api.parse('I feel stomach pain but no coughing today.')
# Log.debug(resp)
if __name__ == '__main__':
FLAGS = argparse.ArgumentParser(
epilog='Infermedica diagnosis API.\n',
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
description='Test infermedica API'
)
FLAGS.add_argument('-id', '--id', type=str, default='',
help='Your infermedica APP ID.')
FLAGS.add_argument('-k', '--key', type=str, default='',
help='Your infermedica API Key.')
FLAGS.add_argument('-m', '--mode', type=bool, default=True,
help='Make requests in development mode (Only during development).')
args = FLAGS.parse_args()
# Display command-line arguments & their values.
Log.info('\n{0}\n{1:^55}\n{0}'
.format('-' * 55, 'Command Line Arguments'))
for k, v in vars(args).items():
Log.info('{:<20} = {!s:>20}'.format(k, v))
Log.info('\n{}\n'.format('-' * 55))
main()
gpt2_input = self._get_gpt2_inputs(questions[0], topk_question,
topk_answer)
gpt2_pred = self.estimator.predict(
lambda: gpt2_estimator.predict_input_fn(inputs=gpt2_input,
batch_size=self.batch_size,
checkpoint_path=self.
gpt2_weight_file))
raw_output = gpt2_estimator.predictions_parsing(
gpt2_pred, self.encoder)
# result_list = [re.search('`ANSWER:(.*)`QUESTION:', s)
# for s in raw_output]
# result_list = [s for s in result_list if s]
# try:
# r = result_list[0].group(1)
# except (AttributeError, IndexError):
# r = topk_answer[0]
refine1 = re.sub('`QUESTION:.*?`ANSWER:',
'',
str(raw_output[0]),
flags=re.DOTALL)
refine2 = refine1.split('`QUESTION: ')[0]
return refine2
if __name__ == "__main__":
from config.util import Log
gen = GenerateQADoc()
Log.info(gen.predict('my eyes hurt'))