Spaces:
Build error
Build error
Update backend_utils.py
Browse files- 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,
|