Update src/app.py

src/app.py

@@ -1,9 +1,21 @@
+import os
+import sys
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import nltk
 import math
 import torch
 
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
+os.environ['HF_HOME'] = '/tmp/hf_cache'
+
+os.environ['NLTK_DATA'] = '/tmp/nltk_data'
+
+cache_paths = ['/tmp/hf_cache', '/tmp/nltk_data']
+for path in cache_paths:
+    os.makedirs(path, exist_ok=True)
+
+
 model_name = "dnj0/t5_base_article_sum"
 max_input_length = 512