Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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('
|
| 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)
|