NagaSaiAbhinay commited on
Commit
9b25486
·
verified ·
1 Parent(s): 4198589

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ datasets:
4
+ - tomg-group-umd/ContraStyles
5
+ library_name: transformers
6
+ ---
7
+ An unofficial implementation of [CSD](https://github.com/learn2phoenix/CSD)
8
+
9
+ Inspired by [vvmatorin/CSD](https://huggingface.co/vvmatorin/CSD), the difference in this implementation is that the CLIP backbone is not the openai clip class but an instance of `CLIPVisionModel`.
10
+
11
+ Inference:
12
+
13
+ ```python
14
+ from PIL import Image
15
+ from transformers import AutoProcessor, AutoModel
16
+
17
+ model = AutoModel.from_pretrained("NagaSaiAbhinay/CSD", trust_remote_code=True)
18
+ processor = AutoProcessor.from_pretrained("NagaSaiAbhinay/CSD")
19
+
20
+ img = Image.open('test_image.png')
21
+ pixel_values = processor
22
+
23
+ ```