update app

app.py

@@ -3,7 +3,7 @@ import yfinance as yf
 import pandas as pd
 import numpy as np
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 from datetime import datetime, timedelta
 import plotly.graph_objects as go
 import plotly.express as px
@@ -30,10 +30,11 @@ 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 = AutoModelForCausalLM.from_pretrained(
+    model = AutoModelForSeq2SeqLM.from_pretrained(
         "amazon/chronos-bolt-base",
         torch_dtype=torch.bfloat16,
-        device_map="auto"
+        device_map="auto",
+        trust_remote_code=True
     )
     tokenizer = AutoTokenizer.from_pretrained("amazon/chronos-bolt-base")
     return model, tokenizer