| 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("تم تحويل النموذج بنجاح!") | |