Update app.py

app.py

@@ -240,7 +240,7 @@ if st.session_state.screen == 2:
 
         # show character locations
         class_names = {0: "Waldo", 1: "Wenda", 2: "Odlaw", 3: "Wizard", 4: "Woof"}
-        for idx, pred in enumerate(results):
+        for idx, pred in enumerate(st.session_state.results):
             #extract detected character names and their confidence scores
             st.session_state.detected_characters = {}
             for *_, cls_id, conf in pred:
@@ -252,7 +252,7 @@ if st.session_state.screen == 2:
         # display locations
         st.session_state.locations = ""
         for name, idx in st.session_state.detected_characters.items():
-            st.session_state.locations += f" {name} page: {str(idx)}, )"
+            st.session_state.locations += f" {name} tiles: {str(idx)}, )"
         st.write(st.session_state.locations)
             
         # plot predicted image