update models 1.4

models.py

@@ -1,7 +1,7 @@
 import torch
 import numpy as np
 import pandas as pd
-from chronos.pipeline import ChronosPipeline
+from chronos import BaseChronosPipeline # GANTI DARI 'chronos.pipeline import ChronosPipeline'
 import warnings
 warnings.filterwarnings('ignore')
 
@@ -16,7 +16,7 @@ def load_timesfm_model():
         try:
             print(f"Loading Chronos Pipeline: {model_name}")
             # ChronosPipeline handles model, tokenizer, and device placement automatically
-            _pipeline = ChronosPipeline.from_pretrained(
+            _pipeline = BaseChronosPipeline.from_pretrained( # MENGGUNAKAN BaseChronosPipeline
                 model_name,
                 # Setting device_map="auto" to automatically select available device
                 device_map="auto"
@@ -24,7 +24,7 @@ def load_timesfm_model():
             print(f"Model loaded successfully.")
             return _pipeline
         except Exception as e:
-            print(f"Error loading model: {e}. Please ensure the 'chronos' library is installed.")
+            print(f"Error loading model: {e}. Please ensure the 'chronos-forecasting' library is installed.")
             _pipeline = None
             return None
     return _pipeline