aatu18 commited on
Commit
b374f9a
·
verified ·
1 Parent(s): 7217cf3
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -336,6 +336,7 @@ def construct_and_execute_api_call(query):
336
  initial_cancer_entities = return_initial_cancer_entities(
337
  query, model="en_ner_bc5cdr_md"
338
  )
 
339
  except Exception as e:
340
  print("unable to guess cancer entities {}".format(str(e)))
341
  initial_cancer_entities = []
@@ -356,6 +357,8 @@ def construct_and_execute_api_call(query):
356
  print("gene entities {}".format(gene_entities))
357
  print("mutation entities {}".format(mutation_entities))
358
  print("cancer entities {}".format(cancer_entities))
 
 
359
 
360
  # infer user intent
361
  intent = infer_user_intent(query)
 
336
  initial_cancer_entities = return_initial_cancer_entities(
337
  query, model="en_ner_bc5cdr_md"
338
  )
339
+ print('initial cancer entities {}'.format(initial_cancer_entities))
340
  except Exception as e:
341
  print("unable to guess cancer entities {}".format(str(e)))
342
  initial_cancer_entities = []
 
357
  print("gene entities {}".format(gene_entities))
358
  print("mutation entities {}".format(mutation_entities))
359
  print("cancer entities {}".format(cancer_entities))
360
+ # only for debug
361
+ cancer_entities = cancer_entities[:2]
362
 
363
  # infer user intent
364
  intent = infer_user_intent(query)