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)
|