modified

app.py

@@ -30,8 +30,8 @@ tokenizer, model = None, None
 try:
     # Attempt to load the tokenizer and model
     with st.spinner("Loading model and tokenizer..."):
-        tokenizer = AutoTokenizer.from_pretrained(model_name, token=True)
-        model = AutoModelForCausalLM.from_pretrained(model_name, token=True)
+        tokenizer = AutoTokenizer.from_pretrained(model_name, token=True, cache_dir="./cache")
+        model = AutoModelForCausalLM.from_pretrained(model_name, token=True, cache_dir="./cache")
     st.success("Model and tokenizer loaded successfully!")
 except Exception as e:
     st.error("Failed to load the model or tokenizer. Please check your configuration.")