Update app.py

app.py

@@ -6,6 +6,10 @@ import yfinance as yf
 import pandas as pd
 import pandas_ta as ta
 import torch
+import warnings
+
+# Suppress specific FutureWarning
+warnings.filterwarnings("ignore", category=FutureWarning)
 
 # Configuration
 ALPACA_API_KEY = 'AKRII2NASCQ4UYYNMUBJ'
@@ -22,7 +26,7 @@ api = REST(ALPACA_API_KEY, ALPACA_SECRET_KEY, ALPACA_BASE_URL)
 
 # Load Models
 dolly_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-dolly_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16) # half precision
+dolly_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, force_download=True) # half precision
 dolly_model.gradient_checkpointing_enable() # memory-efficient loading
 print("Loading FinBERT model...")
 finbert_tokenizer = AutoTokenizer.from_pretrained(FINBERT_MODEL_NAME)