small fix
This commit is contained in:
parent
e5a7a975e8
commit
566a8cd868
@ -16,7 +16,7 @@ def image_classification(path, model):
|
|||||||
|
|
||||||
predictions = model.predict(img_array)
|
predictions = model.predict(img_array)
|
||||||
score = tf.nn.softmax(predictions[0])
|
score = tf.nn.softmax(predictions[0])
|
||||||
print(class_names[np.argmax(score)])
|
# print(class_names[np.argmax(score)])
|
||||||
return class_names[np.argmax(score)]
|
return class_names[np.argmax(score)]
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user