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,
|