Upload folder using huggingface_hub

demo_cpu.py

@@ -42,13 +42,13 @@ def load_model():
     """Load the model and tokenizer."""
     logger.info("Loading model on CPU...")
     model = AutoModelForCausalLM.from_pretrained(
-        "../boolean_model_llama/merged_llama",
+        "Zwounds/boolean-search-model",
         torch_dtype=torch.float32,
         device_map="cpu",
         trust_remote_code=True
     )
     tokenizer = AutoTokenizer.from_pretrained(
-        "../boolean_model_llama/merged_llama",
+        "Zwounds/boolean-search-model",
         trust_remote_code=True
     )