Update app.py

app.py

@@ -157,7 +157,7 @@ class KidneyCNN(nn.Module):
   
   
 if st.session_state.framework =="PyTorch":
-    st.session_state.model = torch.load('models/kidney_model .pth', map_location=torch.device('cpu'))
+    st.session_state.model = torch.load('kidney_model .pth', map_location=torch.device('cpu'))
     st.session_state.model.eval()
     print(type(st.session_state.model)) 
     
@@ -281,7 +281,7 @@ if st.session_state.framework == "PyTorch":
 
         return prob_class_0, prob_class_1, probs
 else:
-    logo_path = "images/tensorflow.png"
+    logo_path = "tensorflow.png"
     bg_color = "orange"  # For example, a warm red/orange
     bg_color_iv = "#FF5733"  # For example, a warm red/orange
 
@@ -1382,7 +1382,7 @@ if page == "pome":
         if st.session_state.show_desc:
         # components.html(html_string)  # JavaScript works
             # st.markdown(html_string, unsafe_allow_html=True)
-            image_path = "images/image.jpg"
+            image_path = "image.jpg"
 
             st.container()
             st.markdown(