Upload folder using huggingface_hub

demo.py

@@ -39,11 +39,11 @@ def load_model():
     """Load the model and set up tokenizer."""
     logger.info("Loading model...")
     model = AutoModelForCausalLM.from_pretrained(
-        "Zwounds/boolean_model_merged",
+        "Zwounds/boolean-search-model",
         device_map="auto",
         torch_dtype=torch.float16
     )
-    tokenizer = AutoTokenizer.from_pretrained("boolean_model_merged")
+    tokenizer = AutoTokenizer.from_pretrained("Zwounds/boolean-search-model")
     tokenizer.use_default_system_prompt = False
     logger.info("Model loaded successfully")