Upload model.py with huggingface_hub

model.py

@@ -1048,6 +1048,11 @@ class Interfuser(nn.Module):
         traffic = self.traffic_pred_head(traffic_feature_with_vel)
         return traffic, waypoints, is_junction, traffic_light_state, stop_sign, traffic_feature
 
+# --- Model Builder Function ---
+# model.py
+
+# ... (كل الكود الآخر في الملف يبقى كما هو) ...
+
 # --- Model Builder Function ---
 @register_model
 def interfuser_baseline(**kwargs):
@@ -1057,8 +1062,12 @@ def interfuser_baseline(**kwargs):
         embed_dim=256,
         rgb_backbone_name="r50",
         lidar_backbone_name="r18",
-        waypoints_pred_head="gru-command", # Matching the original code's logic
+        # ==================  التعديل الأول ==================
+        # تم تغييرها من "gru-command" إلى "gru" لتطابق الملف المحفوظ
+        waypoints_pred_head="gru", 
+        # ==================  التعديل الثاني ==================
         use_different_backbone=True,
-        direct_concat=False, # Matching the original code's logic
+        # تم تغييرها من False إلى True لتطابق الملف المحفوظ (12 قناة إدخال)
+        direct_concat=True, 
     )
     return model