Update app.py

app.py

@@ -247,7 +247,7 @@ def load_models(model_paths):
     # Load Cargo Vessel model
     logging.info("Loading Cargo Vessel model...")
     # Cargo Vessel model input dimension
-    cargo_in_dim = 12 + 3  # 13 features (without 'year') + 3 future features ('day', 'month', 'time_decimal')
+    cargo_in_dim = 6 + 3  # + 3 future features ('day', 'month', 'time_decimal')
     cargo_model = LSTMModelTeacher(in_dim=cargo_in_dim, hidden_dim=200, forecast_horizon=1, n_layers=10, dropout=0.2)
     cargo_model.load_state_dict(torch.load(model_paths['cargo_vessel'], map_location=torch.device('cpu')))
     cargo_model.eval()
@@ -440,8 +440,7 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
         # Adjust features_to_scale based on the selected model
         if selected_model_name == 'Cargo_Vessel':
             features_to_scale = [
-                "mmsi", "sog_kt", "latitude_degrees", "longitude_degrees", "cog_degrees",
-                "dimension_a_m", "dimension_b_m", "dimension_c_m", "dimension_d_m",
+                "mmsi", "latitude_degrees", "longitude_degrees",
                 "day", "month", "time_decimal"  # Removed 'ship_type'
             ]
             future_features = ['day', 'month', 'time_decimal']
@@ -703,7 +702,7 @@ def main():
         'student_north': 'LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_North/horizon1_data_LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_North_seq_24/run_1/best_model.pth',
         'student_mid': 'LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_Mid/horizon1_data_LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_Mid_seq_24/run_1/best_model.pth',
         'student_south': 'LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_South/horizon1_data_LSTM_whole_atlantic_horizon1_with_time_decimal_input_batch256_KD_South_seq_24/run_1/best_model.pth',
-        'cargo_vessel': 'Cago_final/LSTMModel_cargo_horizon1_with_month_day_time_input_batch256/horizon_data_LSTMModel_cargo_horizon1_with_month_day_time_input_batch256_seq_24/run_1/best_model.pth' 
+        'cargo_vessel': 'LSTMModel_cargo_horizon1_with_month_day_time_input_batch256_cleaned/horizon_data_LSTMModel_cargo_horizon1_with_month_day_time_input_batch256_cleaned_seq_24/run_1/best_model.pth' 
     }
 
     scaler_paths = {
@@ -711,7 +710,7 @@ def main():
         'Student_North': 'scaler_train_North_up.joblib',
         'Student_Mid': 'scaler_train_Mid_up.joblib',
         'Student_South': 'scaler_train_South_up.joblib',
-        'Cargo_Vessel': 'scaler_features_cargo_up_final.joblib'  # Add this line
+        'Cargo_Vessel': 'scaler_features_cargo_cleaned.joblib' 
     }