Update app.py

app.py

@@ -71,7 +71,7 @@ def return_num_columns(img):
     img_in[0, :, :, 1] = img_1ch[:, :]
     img_in[0, :, :, 2] = img_1ch[:, :]
               
-    label_p_pred = model.predict(img_in, verbose=0)
+    label_p_pred = model_classifier.predict(img_in, verbose=0)
     num_col = np.argmax(label_p_pred[0]) + 1
     return num_col