Update app.py
Browse files
app.py
CHANGED
|
@@ -149,8 +149,7 @@ if uploaded_file is not None:
|
|
| 149 |
pred = np.rot90(pred, -j)
|
| 150 |
y_pred += pred / 4
|
| 151 |
|
| 152 |
-
|
| 153 |
-
|
| 154 |
pred = np.where(y_pred > threshold, y_pred, 0)
|
| 155 |
|
| 156 |
plot_prediction(pred)
|
|
|
|
| 149 |
pred = np.rot90(pred, -j)
|
| 150 |
y_pred += pred / 4
|
| 151 |
|
| 152 |
+
if detect or bool(threshold):
|
|
|
|
| 153 |
pred = np.where(y_pred > threshold, y_pred, 0)
|
| 154 |
|
| 155 |
plot_prediction(pred)
|