Update model_handler.py

model_handler.py

@@ -15,7 +15,6 @@ class ModelHandler:
         try:
             print(f"Loading {self.model_name} on {self.device}...")
             
-            # Pemuatan otomatis oleh pipeline (sudah terbukti berhasil di langkah sebelumnya)
             self.pipeline = BaseChronosPipeline.from_pretrained(
                 self.model_name,
                 device_map=self.device,
@@ -52,12 +51,20 @@ class ModelHandler:
             predictions_samples = self.pipeline.predict(
                 data['original'],
                 prediction_length=horizon,
-                # KOREKSI: Mengganti 'num_samples' menjadi 'n_samples'
-                n_samples=20 
+                # FIX UTAMA: Menghapus 'n_samples' untuk menghindari error.
+                # Model akan kembali ke single trajectory prediction (deterministic)
             )
             
-            # Mengambil nilai rata-rata (mean) dari semua sampel
-            mean_predictions = np.mean(predictions_samples, axis=0)
+            # Karena sampling dihilangkan, asumsikan output adalah single trajectory (1D atau 2D dengan dimensi pertama 1)
+            if predictions_samples.ndim > 1 and predictions_samples.shape[0] > 1:
+                # Jika model tetap mengembalikan multiple samples (probabilistic)
+                mean_predictions = np.mean(predictions_samples, axis=0)
+            elif predictions_samples.ndim > 1 and predictions_samples.shape[0] == 1:
+                # Jika hanya satu trajectory
+                mean_predictions = predictions_samples[0]
+            else:
+                # Jika sudah 1D
+                mean_predictions = predictions_samples
             
             return mean_predictions