prithivMLmods commited on
Commit
d9a9a7c
·
verified ·
1 Parent(s): 69e77f7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -10,8 +10,11 @@ tags:
10
  - deepfake
11
  ---
12
  # **Deepfake-Quality-Assess-Siglip2**
13
- Deepfake-Quality-Assess-Siglip2 is an image classification vision-language encoder model fine-tuned from `google/siglip2-base-patch16-224`. It is designed for a single-label classification task and classifies the quality of deepfake images.
14
 
 
 
 
15
 
16
  ```python
17
  !pip install -q transformers torch pillow gradio
 
10
  - deepfake
11
  ---
12
  # **Deepfake-Quality-Assess-Siglip2**
13
+ Deepfake-Quality-Assess-Siglip2 is an image classification vision-language encoder model fine-tuned from **google/siglip2-base-patch16-224** for a single-label classification task. It classifies the quality of deepfake images using the **SiglipForImageClassification** architecture.
14
 
15
+ **id2label Mapping:**
16
+ - **0**: "Issue in Deepfake"
17
+ - **1**: "High-Quality Deepfake"
18
 
19
  ```python
20
  !pip install -q transformers torch pillow gradio