thefish1 commited on
Commit
c66b120
·
1 Parent(s): 2fd3098
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -323,8 +323,10 @@ def get_candidates_from_db(keywords_dict, class_name,limit=3):
323
  .with_additional(['distance'])
324
  .do()
325
  )
326
-
327
- if class_name.capitalize() in response['data']['Get']:
 
 
328
  results = response['data']['Get'][class_name.capitalize()]
329
  for result in results:
330
  candidate_list.append({
@@ -332,7 +334,7 @@ def get_candidates_from_db(keywords_dict, class_name,limit=3):
332
  'summary': result['summary'],
333
  'keywords': result['keywords']
334
  })
335
-
336
  return candidate_list
337
 
338
 
 
323
  .with_additional(['distance'])
324
  .do()
325
  )
326
+ class_name=class_name[0].upper()+class_name[1:]
327
+
328
+ if class_name in response['data']['Get']:
329
+ print(111)
330
  results = response['data']['Get'][class_name.capitalize()]
331
  for result in results:
332
  candidate_list.append({
 
334
  'summary': result['summary'],
335
  'keywords': result['keywords']
336
  })
337
+ print(candidate_list)
338
  return candidate_list
339
 
340