diff --git a/serve.py b/serve.py index 853fc1e..868a09e 100644 --- a/serve.py +++ b/serve.py @@ -194,6 +194,7 @@ class MyHandler(BaseHTTPRequestHandler): test_X, test_Y = np.array([[job, gpu_model, time],[1,1,1]]) #data_gainer.get_test_data(return_label_data=True) print(test_X, test_Y) pred_result = predict(config, test_X) + print(pred_result) draw(config, data_gainer, pred_result) msg = {'code': 1, 'error': "container not exist"} except Exception as e: