Update app.py

app.py

@@ -31,7 +31,7 @@ embedding_model = SentenceTransformer("sentence-transformers/paraphrase-multilin
 # Load Jais model and tokenizer
 model_name = "inceptionai/jais-13b"
 tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN)
+model = AutoModelForCausalLM.from_pretrained(model_name, token=HF_TOKEN, trust_remote_code=True)
 
 # RAG function
 def get_response(query, k=3):