Update app.py

app.py

@@ -30,7 +30,7 @@ main_bg = 'download (3).jfif'
 class_labels=[ 'Cyst', 'Normal','Stone', 'Tumor']
 def load_tensorflow_model():
     # Example: Load a pre-trained model (e.g., MobileNetV2)
-    tf_model = tf.keras.models.load_model('model/best_model.keras')
+    tf_model = tf.keras.models.load_model('best_model.keras')
     return tf_model
 def predict_image(image):
     time.sleep(2)  
@@ -123,7 +123,7 @@ def predict_with_pytorch(image):
 
 def load_pytorch_model():
     # Example: Load a pre-trained model (e.g., ResNet18)
-    model = torch.load('model/torch_model.pth', map_location=torch.device('cpu'))  # Forces the model to load on CPU
+    model = torch.load('torch_model.pth', map_location=torch.device('cpu'))  # Forces the model to load on CPU
 
     model.eval()
     return model
@@ -543,7 +543,7 @@ with st.container(key="content-container-3"):
             st.session_state.choice = "pytorch"
 with st.container(key="content-container-1"):
 
-    image_path = "images/t.jpg"
+    image_path = "t.jpg"
     col1, col2 = st.columns([1, 9])
     with col1:
        st.write("")