Spaces:

SBB
/

eynollah-demo

Runtime error

vahidrezanezhad commited on Apr 23

Commit

af99749

•

1 Parent(s): d6d5054

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
     num_col = np.argmax(label_p_pred[0]) + 1
     return num_col

     img_in[0, :, :, 1] = img_1ch[:, :]
     img_in[0, :, :, 2] = img_1ch[:, :]
+    label_p_pred = model_classifier.predict(img_in, verbose=0)
     num_col = np.argmax(label_p_pred[0]) + 1
     return num_col