pemujo commited on
Commit
a9b9ef1
·
1 Parent(s): 5e3a492

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def predict(ash_image, model=model):
77
  im_mean = round(im.mean(),4)
78
  print(im_mean)
79
  if im_mean in images_means:
80
- im = np.load('images/' + images_means[im_mean])
81
 
82
  y_pred = model.predict(im.reshape(1,256, 256, 3))
83
  prediction = np.argmax(y_pred[0], axis=2).reshape(256,256)
 
77
  im_mean = round(im.mean(),4)
78
  print(im_mean)
79
  if im_mean in images_means:
80
+ im = np.load('numpy_files/' + images_means[im_mean])
81
 
82
  y_pred = model.predict(im.reshape(1,256, 256, 3))
83
  prediction = np.argmax(y_pred[0], axis=2).reshape(256,256)