| import torch | |
| from transformers import AutoConfig | |
| # ุชุญู ูู ุงููู ูุฐุฌ ุงูุฃุตูู | |
| model_path = "best_model_improved.pth" | |
| model = torch.load(model_path) | |
| # ุฅูุดุงุก ู ูู ุงูุชูููู ุฅุฐุง ูู ููู ู ูุฌูุฏุงู | |
| config = AutoConfig.for_model("efficientnet-b0", | |
| num_labels=2, # ุนุฏุฏ ุงููุฆุงุช (ุญูููู/ู ุฒูู) | |
| label2id={"real": 0, "fake": 1}, | |
| id2label={0: "real", 1: "fake"}) | |
| config.save_pretrained("./") | |
| # ุญูุธ ุงููู ูุฐุฌ ุจุตูุบุฉ pytorch_model.bin | |
| torch.save(model.state_dict(), "pytorch_model.bin") | |
| print("ุชู ุชุญููู ุงููู ูุฐุฌ ุจูุฌุงุญ!") | |