Update phishing_email_detection_gpt2.py

Add a test that we can load the model and run data through it. May need dummy data reformatted.

Author: david-thrower

phishing_email_detection_gpt2.py

@@ -545,3 +545,40 @@ def from_config(cls, config):
 # Save the model with custom objects
 gpt_baseline_model.save('gpt_baseline_model.h5', save_format='h5', custom_objects=custom_objects)
 cerebros_base_model.save('cerebros_base_model.h5', save_format='h5', custom_objects=custom_objects)
+
+# Test loading the models back
+print("Testing model loading...")
+try:
+    # Load GPT baseline model
+    loaded_gpt_model = tf.keras.models.load_model('gpt_baseline_model.h5', custom_objects=custom_objects)
+    print("✓ GPT baseline model loaded successfully!")
+    
+    # Verify GPT model structure
+    print("GPT Model Summary:")
+    print(loaded_gpt_model.summary())
+    
+    # Test GPT model prediction
+    test_input = tf.constant(["This is a test email for phishing detection."])
+    gpt_prediction = loaded_gpt_model.predict(test_input)
+    print(f"GPT Model prediction shape: {gpt_prediction.shape}")
+    print(f"GPT Model prediction sample: {gpt_prediction[0]}")
+    
+    # Load Cerebros base model
+    loaded_cerebros_model = tf.keras.models.load_model('cerebros_base_model.h5', custom_objects=custom_objects)
+    print("✓ Cerebros base model loaded successfully!")
+    
+    # Verify Cerebros model structure
+    print("Cerebros Model Summary:")
+    print(loaded_cerebros_model.summary())
+    
+    # Test Cerebros model prediction
+    cerebros_prediction = loaded_cerebros_model.predict(test_input)
+    print(f"Cerebros Model prediction shape: {cerebros_prediction.shape}")
+    print(f"Cerebros Model prediction sample shape: {cerebros_prediction[0].shape}")
+    
+    print("✓ All models loaded and validated successfully!")
+    
+except Exception as e:
+    print(f"✗ Error loading models: {e}")
+    raise
+