Update app.py

app.py

@@ -4,6 +4,13 @@ import pandas as pd
 import re
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import subprocess
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/home/user/.cache/huggingface"
+os.environ["HF_HOME"] = "/home/user/.cache/huggingface"
+os.environ["TORCH_HOME"] = "/home/user/.cache/torch"
+os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
+
+import torch
 
 # Set Streamlit page config
 st.set_page_config(page_title="FinBrief: Financial Document Insights", layout="wide")