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)
|
||||
score = tf.nn.softmax(predictions[0])
|
||||
print(class_names[np.argmax(score)])
|
||||
# print(class_names[np.argmax(score)])
|
||||
return class_names[np.argmax(score)]
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user