Update app.py

app.py

@@ -75,14 +75,16 @@ model.summary()
 
 
 
-img = load_img(uploaded_file, grayscale=True, target_size=(28, 28))
-img = img_to_array(img)
-img = img.reshape(1, 1, 28, 28)
-img = img.astype('float32')
-img = img / 255.0
-
-predict_x = model.predict(img)
-classes_x = np.argmax(predict_x,axis=1)
-
+if uploaded_file is not None:
 
-st.write(classes_x[0])
+    img = load_img(uploaded_file, grayscale=True, target_size=(28, 28))
+    img = img_to_array(img)
+    img = img.reshape(1, 1, 28, 28)
+    img = img.astype('float32')
+    img = img / 255.0
+    
+    predict_x = model.predict(img)
+    classes_x = np.argmax(predict_x,axis=1)
+    
+    sign = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y']
+    st.write(sign[classes_x[0]])