Update phishing_email_detection_gpt2.py

david-thrower · Jun 13, 2024 · 516e3f6 · 516e3f6
1 parent f1aaab1
commit 516e3f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -29,7 +29,7 @@
     import zero_7_exp_decay, zero_95_exp_decay, simple_sigmoid
 from ast import literal_eval
 
-from custom.custom import GPT2Layer
+from cerebros.custom.custom import GPT2Layer
 
 #
 # Load the email data