Update app.py

app.py

@@ -52,11 +52,14 @@ st.set_option("deprecation.showfileUploaderEncoding", False)
 @st.cache_resource(show_spinner=False)
 def load_model():
     try:
-        from huggingface_hub import from_pretrained_keras, hf_hub_download
+        from huggingface_hub import hf_hub_download
+        from keras.layers import TFSMLayer
         
         # Download the model files directly
         model_path = hf_hub_download(repo_id="ryefoxlime/PneumoniaDetection", filename="saved_model.pb")
-        model = keras.models.load_model(model_path)
+        
+        # Use TFSMLayer to load the model
+        model = TFSMLayer(model_path, call_endpoint='serving_default')
         return model
     except Exception as e:
         st.error(f"Error loading model: {str(e)}")