Update app.py

app.py

@@ -4,11 +4,21 @@ import cv2
 from flask import Flask, request, jsonify, render_template
 from tensorflow.keras.models import load_model  # type: ignore
 
-# Specify the template folder manually
-app = Flask(__name__, template_folder='.')
+app = Flask(__name__)
+
+# Path to the model file
+model_path = 'pneumonia_model_final.keras'
+
+# Check if the model file exists
+if not os.path.exists(model_path):
+    raise FileNotFoundError(f"The model file {model_path} does not exist.")
 
 # Load the saved model
-model = load_model('pneumonia_model_final.keras')
+try:
+    model = load_model(model_path)
+except ValueError as e:
+    print(f"Error loading the model: {e}")
+    raise
 
 def preprocess_image(image):
     """Preprocess the image for prediction."""