Update inference.py

inference.py

@@ -9,8 +9,11 @@ from nets.model import Model
 from Actor.actor import Actor
 from dataloader import VRP_Dataset
 
+# --- تحديد المجلد الآمن ---
+safe_data_dir = "/home/user/data"
+
 # --- تحميل الإعدادات ---
-params_path = '/data/params_saved.json'
+params_path = os.path.join(safe_data_dir, 'params_saved.json')
 if not os.path.exists(params_path):
     raise FileNotFoundError(f"Settings file not found at {params_path}")
 
@@ -22,14 +25,14 @@ device = params['device']
 dataset_path = params['dataset_path']
 
 # --- مسار النموذج المحفوظ ---
-model_path = "/data/model_state_dict.pt"
+model_path = os.path.join(safe_data_dir, "model_state_dict.pt")
 if not os.path.exists(model_path):
     raise FileNotFoundError(f"Model not found at {model_path}")
 
 # --- تحضير مجلد النتائج ---
-os.makedirs("/data", exist_ok=True)
-txt_results_file = "/data/inference_results.txt"
-csv_results_file = "/data/inference_results.csv"
+os.makedirs(safe_data_dir, exist_ok=True)
+txt_results_file = os.path.join(safe_data_dir, "inference_results.txt")
+csv_results_file = os.path.join(safe_data_dir, "inference_results.csv")
 
 # --- إعداد بيانات inference ---
 inference_dataset = VRP_Dataset(