Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -231,7 +231,7 @@ def detect_fault(uploaded_image):
|
|
| 231 |
|
| 232 |
y_score_image = s_star.cpu().numpy()
|
| 233 |
y_pred_image = 1*(y_score_image >= best_threshold)
|
| 234 |
-
class_label = ['Image
|
| 235 |
|
| 236 |
# --- Plot results ---
|
| 237 |
fig, axs = plt.subplots(1, 3, figsize=(15, 5))
|
|
|
|
| 231 |
|
| 232 |
y_score_image = s_star.cpu().numpy()
|
| 233 |
y_pred_image = 1*(y_score_image >= best_threshold)
|
| 234 |
+
class_label = ['Image Is OK','Image is Not OK']
|
| 235 |
|
| 236 |
# --- Plot results ---
|
| 237 |
fig, axs = plt.subplots(1, 3, figsize=(15, 5))
|