ayan4m1 commited on
Commit
784e394
·
1 Parent(s): b032f48

feat: add model and update pipeline_tag to reflect change in task

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +47 -0
  3. model.onnx +3 -0
  4. preprocessor_config.json +24 -0
README.md CHANGED
@@ -4,7 +4,7 @@ language:
4
  - en
5
  base_model:
6
  - prithivMLmods/Watermark-Detection-SigLIP2
7
- pipeline_tag: image-classification
8
  library_name: transformers.js
9
  tags:
10
  - Image-Classification
 
4
  - en
5
  base_model:
6
  - prithivMLmods/Watermark-Detection-SigLIP2
7
+ pipeline_tag: zero-shot-image-classification
8
  library_name: transformers.js
9
  tags:
10
  - Image-Classification
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "architectures": [
4
+ "SiglipForImageClassification"
5
+ ],
6
+ "id2label": {
7
+ "0": "No Watermark",
8
+ "1": "Watermark"
9
+ },
10
+ "initializer_factor": 1.0,
11
+ "label2id": {
12
+ "No Watermark": 0,
13
+ "Watermark": 1
14
+ },
15
+ "model_type": "siglip",
16
+ "problem_type": "single_label_classification",
17
+ "text_config": {
18
+ "attention_dropout": 0.0,
19
+ "hidden_act": "gelu_pytorch_tanh",
20
+ "hidden_size": 768,
21
+ "intermediate_size": 3072,
22
+ "layer_norm_eps": 1e-06,
23
+ "max_position_embeddings": 64,
24
+ "model_type": "siglip_text_model",
25
+ "num_attention_heads": 12,
26
+ "num_hidden_layers": 12,
27
+ "projection_size": 768,
28
+ "torch_dtype": "float32",
29
+ "vocab_size": 256000
30
+ },
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.51.3",
33
+ "vision_config": {
34
+ "attention_dropout": 0.0,
35
+ "hidden_act": "gelu_pytorch_tanh",
36
+ "hidden_size": 768,
37
+ "image_size": 224,
38
+ "intermediate_size": 3072,
39
+ "layer_norm_eps": 1e-06,
40
+ "model_type": "siglip_vision_model",
41
+ "num_attention_heads": 12,
42
+ "num_channels": 3,
43
+ "num_hidden_layers": 12,
44
+ "patch_size": 16,
45
+ "torch_dtype": "float32"
46
+ }
47
+ }
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36ac63709190503b6081024044b66ffd3296712c46be43a466b091c10d1dd099
3
+ size 1501009071
preprocessor_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "SiglipImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "processor_class": "SiglipProcessor",
18
+ "resample": 2,
19
+ "rescale_factor": 0.00392156862745098,
20
+ "size": {
21
+ "height": 224,
22
+ "width": 224
23
+ }
24
+ }