Update app.py

app.py

@@ -38,7 +38,19 @@ class MultiAttributeClassifier:
                 model_path = f"models/classification/{category}_model.h5"
                 if os.path.exists(model_path):
                     print(f"🔍 Loading model: {model_path}")
-                    self.models[category] = tf.keras.models.load_model(model_path)
+                    try:
+                        # Try normal loading first
+                        self.models[category] = tf.keras.models.load_model(model_path)
+                    except Exception as e1:
+                        print(f"   ⚠️ Normal loading failed: {e1}")
+                        try:
+                            # Try with compile=False
+                            self.models[category] = tf.keras.models.load_model(model_path, compile=False)
+                            print(f"   ✅ Loaded {category} with compile=False")
+                        except Exception as e2:
+                            print(f"   ❌ Failed to load {category}: {e2}")
+                            continue
+                    
                     print(f"✅ Loaded {category} model ({os.path.getsize(model_path)/1024/1024:.1f} MB)")
                     
                     # Load encoder
@@ -132,7 +144,30 @@ class MultiAttributeClassifier:
                 try:
                     if os.path.exists(model_path):
                         print(f"🔍 Trying to load: {model_path}")
-                        self.gan_models[model_name] = tf.keras.models.load_model(model_path)
+                        # Try loading with different compatibility options
+                        try:
+                            # First try: Normal loading
+                            self.gan_models[model_name] = tf.keras.models.load_model(model_path)
+                        except Exception as e1:
+                            print(f"   ⚠️ Normal loading failed: {e1}")
+                            try:
+                                # Second try: Load with compile=False (ignore training config)
+                                self.gan_models[model_name] = tf.keras.models.load_model(model_path, compile=False)
+                                print(f"   ✅ Loaded with compile=False")
+                            except Exception as e2:
+                                print(f"   ⚠️ compile=False failed: {e2}")
+                                try:
+                                    # Third try: Load with custom objects (for any custom layers)
+                                    self.gan_models[model_name] = tf.keras.models.load_model(
+                                        model_path, 
+                                        compile=False,
+                                        custom_objects={'tf': tf}
+                                    )
+                                    print(f"   ✅ Loaded with custom_objects")
+                                except Exception as e3:
+                                    print(f"   ❌ All loading methods failed: {e3}")
+                                    raise e3
+                        
                         print(f"✅ Loaded GAN: {model_name} from {model_path}")
                         model_loaded = True
                         break