imamnurby commited on
Commit
1c77037
·
1 Parent(s): 9c121cf

Update backend_utils.py

Browse files
Files changed (1) hide show
  1. backend_utils.py +4 -1
backend_utils.py CHANGED
@@ -433,8 +433,10 @@ def make_predictions(input_query,
433
  Returns:
434
  predictions (list): a list of dictionary containing the prediction details
435
  '''
 
436
  library_ids, library_names = retrieve_libraries(model_retrieval, input_query, db_metadata)
437
-
 
438
  predictions = generate_api_usage_patterns_batch(
439
  model_generative,
440
  tokenizer_generative,
@@ -444,6 +446,7 @@ def make_predictions(input_query,
444
  config.get('num_return_sequences')
445
  )
446
 
 
447
  hw_configs = predict_hw_config(
448
  model_classifier,
449
  tokenizer_classifier,
 
433
  Returns:
434
  predictions (list): a list of dictionary containing the prediction details
435
  '''
436
+ print("retrieve library")
437
  library_ids, library_names = retrieve_libraries(model_retrieval, input_query, db_metadata)
438
+
439
+ print("generate hw patterns")
440
  predictions = generate_api_usage_patterns_batch(
441
  model_generative,
442
  tokenizer_generative,
 
446
  config.get('num_return_sequences')
447
  )
448
 
449
+ print("generate hw config")
450
  hw_configs = predict_hw_config(
451
  model_classifier,
452
  tokenizer_classifier,