Skip to content

Commit 57edcc3

Browse files
Update phishing_email_detection_gpt2.py
Still mismatched var names ....
1 parent 10760a4 commit 57edcc3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

phishing_email_detection_gpt2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -883,7 +883,7 @@ def call(self, inputs):
883883
# token_ids = list(token_ids)
884884

885885
# Now pass the list of integers to your generate method
886-
generated_tokens = reconstituted_generator.generate(
886+
generated_tokens = generator.generate(
887887
token_ids=half_sample_tokenized, # This should now be a list of integers
888888
do_sample=False,
889889
max_new_tokens=40
@@ -993,7 +993,7 @@ def call(self, inputs):
993993
# token_ids = list(token_ids)
994994

995995
# Now pass the list of integers to your generate method
996-
generated_tokens = reconstituted_generator.generate(
996+
generated_tokens = reconstituted_generator.generate(
997997
token_ids=half_sample_tokenized, # This should now be a list of integers
998998
do_sample=False,
999999
max_new_tokens=40

0 commit comments

Comments
 (0)