Upload model.py with huggingface_hub

model.py

@@ -1054,20 +1054,18 @@ class Interfuser(nn.Module):
 # ... (كل الكود الآخر في الملف يبقى كما هو) ...
 
 # --- Model Builder Function ---
-@register_model
-def interfuser_baseline(**kwargs):
+def build_interfuser_model(config):
+    """
+    تبني نموذج Interfuser بناءً على قاموس إعدادات.
+    """
     model = Interfuser(
-        enc_depth=6,
-        dec_depth=6,
-        embed_dim=256,
-        rgb_backbone_name="r50",
-        lidar_backbone_name="r18",
-        # ==================  التعديل الأول ==================
-        # تم تغييرها من "gru-command" إلى "gru" لتطابق الملف المحفوظ
-        waypoints_pred_head="gru", 
-        # ==================  التعديل الثاني ==================
-        use_different_backbone=True,
-        # تم تغييرها من False إلى True لتطابق الملف المحفوظ (12 قناة إدخال)
-        direct_concat=True, 
+        enc_depth=config.get("enc_depth", 6),
+        dec_depth=config.get("dec_depth", 6),
+        embed_dim=config.get("embed_dim", 256),
+        rgb_backbone_name=config.get("rgb_backbone_name", "r50"),
+        lidar_backbone_name=config.get("lidar_backbone_name", "r18"),
+        waypoints_pred_head=config.get("waypoints_pred_head", "gru"), 
+        use_different_backbone=config.get("use_different_backbone", True),
+        direct_concat=config.get("direct_concat", True), 
     )
     return model