Update app.py

app.py

@@ -26,11 +26,8 @@ def download_model():
 # Function to load and save the model in the SavedModel format
 def convert_and_save_model(h5_path, save_dir):
     try:
-        # Load the .h5 model without compile=False
         model = tf.keras.models.load_model(h5_path)
-        
-        # Save the model in .keras format
-        model.save(save_dir)  # No need to specify save_format
+        model.save(save_dir)  # Save in .keras format
         st.success(f"Model saved in .keras format at '{save_dir}'.")
         return model
     except Exception as e:
@@ -47,14 +44,20 @@ def load_saved_model(saved_model_path):
         return None
 
 # Download the model if not already done
+# Check if the model is being downloaded and saved correctly
 download_model()
 
-# Convert and save the model to avoid deserialization issues
+# Convert and save the model if it doesn't exist
 if not os.path.exists(saved_model_dir):
     model = convert_and_save_model(local_h5_path, saved_model_dir)
 else:
     model = load_saved_model(saved_model_dir)
 
+# Check if the model was loaded successfully
+if model is None:
+    st.error("Failed to load the model. Please check the model file.")
+else:
+    st.success("Model loaded successfully.")
 # Define the class labels
 label_map = {0: 'Glioma', 1: 'Meningioma', 2: 'Normal', 3: 'Pituitary'}