Upload folder using huggingface_hub

demo.py

@@ -39,7 +39,7 @@ def load_model():
     """Load the model and set up tokenizer."""
     logger.info("Loading model...")
     model = AutoModelForCausalLM.from_pretrained(
-        "boolean_model_merged",
+        "Zwounds/boolean_model_merged",
         device_map="auto",
         torch_dtype=torch.float16
     )