Update README.md

README.md

@@ -9,9 +9,22 @@ This is a Gemma_2b_en fine-tuned on QA pairs (3.5k) generated from Cosmology and
 from 2018-2022 and tested on QA pairs (1k) generated from 2023 articles, scoring over 75% accuracy.
 
 
+
+
 To generate an answer for a given question using this model, please use:
 
+Requirements:
+
 ```
+keras==3.6.0
+keras_nlp==0.15.1
+```
+import os
+
+os.environ["KERAS_BACKEND"] = "jax"  # Or "torch" or "tensorflow".                                                                                                                          
+# Avoid memory fragmentation on JAX backend.                                                                                                                                                
+os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]="1.00"
+
 import keras
 import keras_nlp