Skip to content

Commit 10760a4

Browse files
Update phishing_email_detection_gpt2.py
Variable name mismatch ...
1 parent 9020bd9 commit 10760a4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

phishing_email_detection_gpt2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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 = 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)