imamnurby commited on
Commit
291c5dd
·
1 Parent(s): 8a67152

Update backend_utils.py

Browse files
Files changed (1) hide show
  1. backend_utils.py +6 -0
backend_utils.py CHANGED
@@ -469,6 +469,10 @@ def make_predictions(input_query,
469
  '''
470
  library_ids, library_names = retrieve_libraries(model_retrieval, input_query, db_metadata)
471
 
 
 
 
 
472
  predictions = generate_api_usage_patterns_batch(
473
  model_generative,
474
  tokenizer_generative,
@@ -480,6 +484,7 @@ def make_predictions(input_query,
480
  config.get('max_length_generate')
481
  )
482
 
 
483
  hw_configs = predict_hw_config(
484
  model_classifier,
485
  tokenizer_classifier,
@@ -492,6 +497,7 @@ def make_predictions(input_query,
492
  for output_dict, hw_config in zip(predictions, hw_configs):
493
  output_dict['hw_config'] = hw_config
494
 
 
495
  predictions = get_metadata_library(predictions, db_metadata)
496
 
497
  return predictions
 
469
  '''
470
  library_ids, library_names = retrieve_libraries(model_retrieval, input_query, db_metadata)
471
 
472
+ if len(library_ids) == 0:
473
+ return {'status': 999}
474
+
475
+ print("generate usage patterns")
476
  predictions = generate_api_usage_patterns_batch(
477
  model_generative,
478
  tokenizer_generative,
 
484
  config.get('max_length_generate')
485
  )
486
 
487
+ print("generate hw configs")
488
  hw_configs = predict_hw_config(
489
  model_classifier,
490
  tokenizer_classifier,
 
497
  for output_dict, hw_config in zip(predictions, hw_configs):
498
  output_dict['hw_config'] = hw_config
499
 
500
+ print("finished the predictions")
501
  predictions = get_metadata_library(predictions, db_metadata)
502
 
503
  return predictions