Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def predict(img) -> Tuple[Dict, float]:
|
|
| 53 |
x=np.count_nonzero(pred_prob)
|
| 54 |
print(x)
|
| 55 |
if x==0:
|
| 56 |
-
pred_labels_and_probs={"Unknown Images found":0}
|
| 57 |
else:
|
| 58 |
pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
|
| 59 |
|
|
|
|
| 53 |
x=np.count_nonzero(pred_prob)
|
| 54 |
print(x)
|
| 55 |
if x==0:
|
| 56 |
+
pred_labels_and_probs={"Unknown Images found...Please Provide Images of Headphone,Mobile or laptop":0}
|
| 57 |
else:
|
| 58 |
pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
|
| 59 |
|