Create convert_model.py

convert_model.py

@@ -0,0 +1,18 @@
+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("تم تحويل النموذج بنجاح!")