Update interface.py

The `preprocessed_input` may not has a `shape` attribute.
This commit is contained in:
yalei 2019-07-14 11:51:31 +08:00 committed by GitHub
parent 0169f73f35
commit 91fd9e4803

View File

@ -136,7 +136,7 @@ class Interface:
Method that calls the relevant method of the model object to make a prediction.
:param preprocessed_input: the preprocessed input returned by the input interface
"""
print(preprocessed_input.shape)
# print(preprocessed_input.shape)
if self.model_type == "sklearn":
return self.model_obj.predict(preprocessed_input)
elif self.model_type == "keras":