Update app.py

app.py

@@ -112,11 +112,23 @@ if file is not None:
         st.image(opencv_image, channels="BGR")
     with col2:
         st.header("Results")
+        
+        if file:
+            name = file.name.split(".")[0]
+        if name in df['image_id'].values: 
+            st.write("True Label: ", df.loc[df['image_id']==name, 'dx'].iloc[0])
+            else:
+                st.write("No match")
+        
+        
         # st.write(file.name)
-        # st.write("True Label",true_file[true_file.image_id==file.name]["dx"])
+        # 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])
 
+
+        
+