Update app.py

app.py

@@ -74,8 +74,8 @@ def do_predict(pil_img: PIL.Image.Image):
 
     # For pretty ranked dict expected by gr.Label
     pretty_dict = {
-        "no_tomato": float(row.get("no_tomato (0)", 0.0)),
-        "tomato": float(row.get("tomato (1)", 0.0)),
+        "no_tomato": float(row.get("no_tomato", 0.0)),
+        "tomato": float(row.get("tomato", 0.0)),
     }
 
     return pretty_dict