Update main.py

main.py

@@ -207,11 +207,14 @@ def predict(payload: PredictRequest):
         # ---------------------------------------------------------------------
         # Run model (raw confidence)
         # ---------------------------------------------------------------------
+        MODEL_PATH = "/app/models/Forest_Segmentation_Best.keras"
+
         model = tf.keras.models.load_model(
-            "models/Forest_Segmentation_Best.keras",
+            MODEL_PATH,
             compile=False
         )
 
+
         confidence_map = model.predict(
             input_tensor, verbose=0
         )[0, :, :, 0]