update app 1.1

app.py

@@ -3,7 +3,7 @@ import yfinance as yf
 import pandas as pd
 import numpy as np
 import torch
-from transformers import AutoModelForTimeSeriesForecasting, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from datetime import datetime, timedelta
 import plotly.graph_objects as go
 import plotly.express as px
@@ -29,7 +29,7 @@ from config import IDX_STOCKS, TECHNICAL_INDICATORS, PREDICTION_CONFIG
 @spaces.GPU(duration=120)
 def load_model():
     """Load the Amazon Chronos-Bolt model for time series forecasting"""
-    model = AutoModelForTimeSeriesForecasting.from_pretrained(
+    model = AutoModelForCausalLM.from_pretrained(
         "amazon/chronos-bolt-base",
         torch_dtype=torch.bfloat16,
         device_map="auto"