Update app.py

app.py

@@ -29,7 +29,7 @@ if file is not None:
     x = np.squeeze(x)
     x  = np.expand_dims(x, axis=0)
     prediction = model.predict(x)
-    index = numpy.argmax(prediction[0])
+    index = np.argmax(prediction[0])
     label = labels[index]
     text = f"<p style='text-align: center;'>{label}</p>"
     col1, col2, col3 = st.columns(3)