Update app.py

app.py

@@ -552,14 +552,14 @@ with st.container(key="content-container-1"):
         st.write("")
         if st.session_state.choice == "tensorflow":
             st.markdown(f""" <div class="title-container">
-        <img src="data:image/png;base64,{base64.b64encode(open("images/tensorflow.png","rb").read()).decode()}" alt="Uploaded Image">
+        <img src="data:image/png;base64,{base64.b64encode(open("tensorflow.png","rb").read()).decode()}" alt="Uploaded Image">
  <h2 class="title">Tensorflow Model Information</h2></div>""", unsafe_allow_html=True)
             st.write("This is a Convolutional Neural Network (CNN) model trained on image data.")
             st.write(f"Input Shape: (64, 64, 3)")
             st.write(f"Output Classes: {4} classes")
         else :
                     st.markdown(f""" <div class="title-container">
-                <img src="data:image/png;base64,{base64.b64encode(open("images/pytorch.png","rb").read()).decode()}" alt="Uploaded Image">
+                <img src="data:image/png;base64,{base64.b64encode(open("pytorch.png","rb").read()).decode()}" alt="Uploaded Image">
         <h2 class="title">Pytorch Model Information</h2></div>""", unsafe_allow_html=True)
                     st.write("This is a Convolutional Neural Network (CNN) model trained on image data.")
                     st.write(f"Input Shape: (64, 64, 3)")