Upload folder using huggingface_hub

example.py

@@ -3,18 +3,8 @@ from PIL import Image, ImageDraw
 from transformers import AutoModel, AutoProcessor, AutoTokenizer, AutoConfig, AutoImageProcessor, GenerationConfig
 from postprocessing import extract_classes_bboxes, transform_bbox_to_original, postprocess_text
 
-from hf_nemotron_parse_config import NemotronParseConfig
-from hf_nemotron_parse_modeling import NemotronParseForConditionalGeneration
-from hf_nemotron_parse_processor import NemotronParseProcessor, NemotronParseImageProcessor
-
-AutoConfig.register("nemotron_parse", NemotronParseConfig)
-AutoModel.register(NemotronParseConfig, NemotronParseForConditionalGeneration)
-AutoProcessor.register("nemotron_parse", NemotronParseProcessor)
-AutoImageProcessor.register("nemotron_parse", NemotronParseImageProcessor)
-
-
 # Load model and processor
-model_path = "." #nvidia/NVIDIA-Nemotron-Parse-v1.1" #Nano-12B-v2-VL-BF16"  # Or use a local path
+model_path = "nvidia/NVIDIA-Nemotron-Parse-v1.1"  # Or use a local path
 device = "cuda:0"
 
 model = AutoModel.from_pretrained(
@@ -26,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
 processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
 
 # Load image
-image = Image.open("../soa_1.png") #path/to/your/image.jpg")
+image = Image.open("path/to/your/image.jpg")
 task_prompt = "</s><s><predict_bbox><predict_classes><output_markdown>"
 
 # Process image

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:81cae06dbfa407fce43e8624cc25a167340eff9e710492e890039605e2ac2570
-size 3827116504
+oid sha256:7469048899485bd828486142c8ac8d4ddd3f3411e9cf6bcc686a73098d403477
+size 3827116472