Update app.py

app.py

@@ -3,15 +3,22 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModel
 
 # Load the model and tokenizer
+@st.cache_resource
+import streamlit as st
+from transformers import AutoTokenizer, AutoModel
+
 @st.cache_resource
 def load_model():
     model_name = "mradermacher/Indian_Legal_Assistant-GGUF"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModel.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
+    model = AutoModel.from_pretrained(model_name, use_auth_token=token)
     return tokenizer, model
 
 tokenizer, model = load_model()
 
+
+tokenizer, model = load_model()
+
 # Streamlit App Layout
 st.title("Indian Legal Assistant - Hugging Face Spaces Deployment")
 st.write("This app provides answers to legal questions using the Indian Legal Assistant model.")