fix: Update model loading to use .h5 format

frontend/app.py

@@ -59,17 +59,14 @@ if 'disease_model' not in st.session_state:
     try:
         # Check if running on Hugging Face Spaces
         if 'HF_SPACE_ID' in os.environ:
-            model_path = 'FINAL_MODEL.zip'
-            with tf.keras.utils.get_file('FINAL_MODEL.keras', model_path, extract=True) as extracted_model_path:
-                model_dir = os.path.dirname(extracted_model_path)
-                model_path = os.path.join(model_dir, 'FINAL_MODEL.keras')
-                st.session_state.disease_model = tf.keras.models.load_model(model_path)
+            model_path = 'FINAL_MODEL.h5'  # Updated to .h5
+            st.session_state.disease_model = tf.keras.models.load_model(model_path)
         else:  # Assume running locally
-            model_path = 'FINAL_MODEL.keras'
+            model_path = 'FINAL_MODEL.h5'  # Updated to .h5
             st.session_state.disease_model = tf.keras.models.load_model(model_path)
         print("Disease model loaded successfully!")
     except FileNotFoundError:
-        st.error("Disease classification model not found. Please ensure 'FINAL_MODEL.zip' (for Hugging Face) or 'FINAL_MODEL.keras' (for local) is in the correct directory.")
+        st.error("Disease classification model not found. Please ensure 'FINAL_MODEL.h5' is in the correct directory.")
         st.session_state.disease_model = None
     except PermissionError:
         st.error("Permission error accessing 'model.weights.h5'. Please ensure the file is not being used by another process.")
@@ -112,16 +109,13 @@ if 'model_llm' not in st.session_state:
 try:
     # Check if running on Hugging Face Spaces
     if 'HF_SPACE_ID' in os.environ:
-        model_path = 'FINAL_MODEL.zip'
-        with tf.keras.utils.get_file('FINAL_MODEL.keras', model_path, extract=True) as extracted_model_path:
-            model_dir = os.path.dirname(extracted_model_path)
-            model_path = os.path.join(model_dir, 'FINAL_MODEL.keras')
-            disease_model = tf.keras.models.load_model(model_path)
+        model_path = 'FINAL_MODEL.h5'  # Updated to .h5
+        disease_model = tf.keras.models.load_model(model_path)
     else:  # Assume running locally
-        model_path = 'FINAL_MODEL.keras'
+        model_path = 'FINAL_MODEL.h5'  # Updated to .h5
         disease_model = tf.keras.models.load_model(model_path)
 except FileNotFoundError:
-    st.error("Disease classification model not found. Please ensure 'FINAL_MODEL.zip' (for Hugging Face) or 'FINAL_MODEL.keras' (for local) is in the correct directory.")
+    st.error("Disease classification model not found. Please ensure 'FINAL_MODEL.h5' is in the correct directory.")
     disease_model = None
 except PermissionError:
     st.error("Permission error accessing 'model.weights.h5'. Please ensure the file is not being used by another process.")