Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -78,8 +78,8 @@ def predict(image: Image.Image):
|
|
| 78 |
sp_conf = float(prob_sp[sp_id].item())
|
| 79 |
st_conf = float(prob_st[st_id].item())
|
| 80 |
|
| 81 |
-
sp_text = f"{SPECIES[sp_id]} (
|
| 82 |
-
st_text = f"{STATE[st_id]} (
|
| 83 |
|
| 84 |
return sp_text, st_text
|
| 85 |
|
|
|
|
| 78 |
sp_conf = float(prob_sp[sp_id].item())
|
| 79 |
st_conf = float(prob_st[st_id].item())
|
| 80 |
|
| 81 |
+
sp_text = f"{SPECIES[sp_id]} (score={sp_conf:.3f})"
|
| 82 |
+
st_text = f"{STATE[st_id]} (score={st_conf:.3f})"
|
| 83 |
|
| 84 |
return sp_text, st_text
|
| 85 |
|