Update app.py

app.py

@@ -17,7 +17,7 @@ def load_llama_model():
     print("Loading base model...")
     model = AutoModelForCausalLM.from_pretrained(
         BASE_MODEL,
-        torch_dtype=torch.bfloat16 if torch.has_bfloat16 else torch.float32,  # Use bfloat16 if available, else float32
+        torch_dtype=torch.float32 
         device_map="cpu",  # Ensure it runs on CPU
         token=HUGGINGFACE_TOKEN
     )