Update model_handler.py

model_handler.py

@@ -1,21 +1,23 @@
 import numpy as np
 import torch
-from chronos import BaseChronosPipeline
+# PENTING: Mengganti ChronosPipeline dengan BaseChronosPipeline sesuai referensi terbaru
+from chronos import BaseChronosPipeline 
 
 class ModelHandler:
     def __init__(self):
         self.model_name = "amazon/chronos-2" 
         self.pipeline = None
+        # Penentuan device
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.load_model()
     
     def load_model(self):
-        """Load Chronos-2 model using the official ChronosPipeline"""
+        """Load Chronos-2 model using the BaseChronosPipeline"""
         try:
             print(f"Loading {self.model_name} on {self.device}...")
             
-            # FIX UTAMA: Pemuatan otomatis oleh pipeline
-            self.pipeline = ChronosPipeline.from_pretrained(
+            # Perhatikan: Menggunakan BaseChronosPipeline.from_pretrained
+            self.pipeline = BaseChronosPipeline.from_pretrained(
                 self.model_name,
                 device_map=self.device,
             )
@@ -49,12 +51,14 @@ class ModelHandler:
                 return np.array(predictions)
             
             # --- Chronos-2 Inference ---
+            # NOTE: BaseChronosPipeline.predict mengembalikan array of arrays (sampel)
             predictions_samples = self.pipeline.predict(
                 data['original'],
                 prediction_length=horizon,
                 num_samples=20 
             )
             
+            # Mengambil nilai rata-rata (mean) dari semua sampel
             mean_predictions = np.mean(predictions_samples, axis=0)
             
             return mean_predictions