Update README.md
Browse files
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
|
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
|