Update app.py

app.py

@@ -20,9 +20,7 @@ import functools
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# ============================
-# Helper Functions
-# ============================
+
 
 def add_time_decimal_feature(df):
     """
@@ -274,9 +272,6 @@ def load_scalers(scaler_paths):
             raise FileNotFoundError(f"Scaler file for {model_name} not found at '{scaler_path}'. Please provide the correct path.")
     return loaded_scalers
 
-# ============================
-# Model Selection Logic
-# ============================
 
 def determine_subarea(df):
     """
@@ -471,7 +466,7 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
 
         logging.info("Input CSV has the correct columns.")
 
-        # Check and add 'time_decimal' if necessary
+        
         if selected_model_name != 'Cargo_Vessel':
             df = add_time_decimal_feature(df)
         else:
@@ -517,7 +512,7 @@ def classical_prediction(file_path, model_choice, min_mmsi, max_mmsi, models, lo
                 all_predictions.append(predictions)
                 all_y_true.append(y_batch.numpy())
 
-        inference_time = time.time() - start_time  # End inference time tracking
+        inference_time = time.time() - start_time  # End inference time 
 
         all_predictions = np.concatenate(all_predictions, axis=0)
         y_true = np.concatenate(all_y_true, axis=0)