"Update space"

Signed-off-by: Emmanuel Frimpong Asante <[email protected]>

app.py

@@ -17,10 +17,10 @@ auth_model = load_model('models/auth_model.h5', compile=True)
 # Initialize the tokenizer
 llama_tokenizer = AutoTokenizer.from_pretrained('meta-llama/Meta-Llama-3-8B-Instruct')
 
-# Initialize the LLaMA model with automatic device allocation
+# Initialize the LLaMA model and force CPU usage
 llama_model = AutoModelForCausalLM.from_pretrained(
     'meta-llama/Meta-Llama-3-8B-Instruct',
-    device_map="auto"  # Automatically allocate the model to the available devices
+    device_map={"": "cpu"}  # Force the model to load on the CPU
 )
 
 # Dictionaries for disease names, results, and recommendations