Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ def predict(img):
|
|
| 14 |
# st.write(learn_inf.predict(img))
|
| 15 |
|
| 16 |
f"""
|
| 17 |
-
## This is **{pred if probs[key].item()>0.
|
| 18 |
### Prediction result: {pred}
|
| 19 |
### Probability of {pred}: {probs[key].item()*100: .2f}%
|
| 20 |
"""
|
|
|
|
| 14 |
# st.write(learn_inf.predict(img))
|
| 15 |
|
| 16 |
f"""
|
| 17 |
+
## This is **{pred if probs[key].item()>0.5 else "not a character"}** of Simpson.
|
| 18 |
### Prediction result: {pred}
|
| 19 |
### Probability of {pred}: {probs[key].item()*100: .2f}%
|
| 20 |
"""
|