Update app.py

app.py

@@ -125,14 +125,15 @@ if file is not None:
             name = file.name.split(".")[0]
         if name in true_file['image_id'].values: 
             st.write("True Label: ", classes_map[true_file.loc[true_file['image_id']==name, 'dx'].iloc[0]])
+            st.write("Prediction:",class_name[1])
         else:
             st.write("No match")
         
         
         # st.write(file.name)
         # st.write("True Label",true_file[true_file.image_id==file.name]["dx"][0])
-        st.write("Prediction",class_name[0])
-        st.metric("Category:", class_name[1])
+        # st.write("Prediction",class_name[0])
+        # st.metric("Category:", class_name[1])