Update model.py

model.py

@@ -38,17 +38,13 @@ LABELS = {}
 def load_models():
     for file in os.listdir(MODELS_DIR):
         name, ext = os.path.splitext(file)
-
-        # Normalize crop key
         crop_name = name.replace("_model", "").lower()
 
         model_path = os.path.join(MODELS_DIR, file)
         label_path = os.path.join(LABELS_DIR, f"{crop_name}_labels.json")
 
         if not os.path.exists(label_path):
-            raise FileNotFoundError(
-                f"Label file missing for {crop_name}: {label_path}"
-            )
+            raise FileNotFoundError(f"Missing labels for {crop_name}")
 
         with open(label_path) as f:
             LABELS[crop_name] = json.load(f)
@@ -62,7 +58,11 @@ def load_models():
             PYTORCH_MODELS[crop_name] = model
 
         elif ext in [".keras", ".h5"]:
-            KERAS_MODELS[crop_name] = tf.keras.models.load_model(model_path)
+            KERAS_MODELS[crop_name] = tf.keras.models.load_model(
+                model_path,
+                custom_objects={"swish": tf.keras.activations.swish},
+                compile=False
+            )
 
 # Load once
 load_models()