Update app.py

app.py

@@ -78,14 +78,18 @@ def predict(image: Image.Image):
     sp_conf = float(prob_sp[sp_id].item())
     st_conf = float(prob_st[st_id].item())
 
-    sp_text = f"{SPECIES[sp_id]}  (id={sp_id}, score={sp_conf:.3f})"
-    st_text = f"{STATE[st_id]}  (id={st_id}, score={st_conf:.3f})"
+    #sp_text = f"{SPECIES[sp_id]}  (id={sp_id}, score={sp_conf:.3f})"
+    #st_text = f"{STATE[st_id]}  (id={st_id}, score={st_conf:.3f})"
+
+    sp_text = f"{SPECIES[sp_id]}  (score={sp_conf:.3f})"
+    st_text = f"{STATE[st_id]}  (score={st_conf:.3f})"
 
     return sp_text, st_text
 
 
 EXAMPLE_TEXT = """
 **Taxa (12):** Cacoxnus indagator, Chelostoma florisomne, Coeliopencyrtus, Eumenidae, Heriades, Hylaeus, Megachile, Osmia bicornis, Osmia cornuta, Passaloecus, Psenulus, Trypoxylon
+
 **States (4):** DauLv, DeadLv, Lv, OldFood
 """
 
@@ -136,10 +140,10 @@ with gr.Blocks(theme=theme, title="LTN EfficientNet Two-Head Classifier") as dem
                 inputs=img,
             )
 
-    gr.Markdown(
-        "<small>Note: The status legend is a human-readable mapping. "
-        "Your model output labels come from `labels.json`.</small>"
-    )
+    #gr.Markdown(
+     #   "<small>Note: The status legend is a human-readable mapping. "
+      #  "Your model output labels come from `labels.json`.</small>"
+    #)
 
 if __name__ == "__main__":
     demo.launch()