Update app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ if uploaded_file is not None:
|
|
80 |
|
81 |
y_pred = 0
|
82 |
for j in [0,1,2,3]:
|
83 |
-
rotated = np.rot90(
|
84 |
pred = model.predict(rotated.reshape(1, 128, 128, 1)).reshape(128 ,128)
|
85 |
pred = np.rot90(pred, -j)
|
86 |
y_pred += pred / 4
|
|
|
80 |
|
81 |
y_pred = 0
|
82 |
for j in [0,1,2,3]:
|
83 |
+
rotated = np.rot90(image_data, j)
|
84 |
pred = model.predict(rotated.reshape(1, 128, 128, 1)).reshape(128 ,128)
|
85 |
pred = np.rot90(pred, -j)
|
86 |
y_pred += pred / 4
|