שלב 4: טעינת המודל המאומן במקום המודל המקורי

לאחר שסיימת לאמן את המודל ושמרת אותו, תוכל לטעון את המודל המאומן באפליקציה שלך במקום המודל המקורי:

from transformers import GPT2LMHeadModel, GPT2Tokenizer

model = GPT2LMHeadModel.from_pretrained('./trained_model')
tokenizer = GPT2Tokenizer.from_pretrained('./trained_model')