Update app.py

app.py

@@ -3,10 +3,10 @@ import gradio as gr
 import psutil
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/data/hf_cache"
 
 tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-120b")
-model = AutoModelForCausalLM.from_pretrained("openai/gpt-oss-120b", device_map="auto", offload_folder="/tmp/offload")
+model = AutoModelForCausalLM.from_pretrained("openai/gpt-oss-120b", device_map="auto", offload_folder="/data/offload")
 
 def get_stats():
     process = psutil.Process(os.getpid())