Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,7 +7,7 @@ from annotated_text import annotated_text
 import warnings
 import pandas as pd
 from pandas import DataFrame
-
+import os
 
 warnings.filterwarnings('ignore')
 import re, flair, random, time
@@ -20,9 +20,13 @@ from flair.models import SequenceTagger
 from flair.trainers import ModelTrainer
 
 from huggingface_hub import hf_hub_download
-# Set environment variable for Hugging Face cache
-os.environ["HF_HOME"] = "./hf_data"  # any writable directory
-os.environ["TRANSFORMERS_CACHE"] = "./hf_data"
+# Use a cache folder inside the app folder
+cache_dir = "./hf_cache"  # relative to your app folder
+os.makedirs(cache_dir, exist_ok=True)
+
+# Tell Hugging Face to use this folder
+os.environ["HF_HOME"] = cache_dir
+os.environ["TRANSFORMERS_CACHE"] = cache_dir
 
 st.set_page_config(
    page_title="Marathi POS Tagger",