Image Classification
Transformers
Safetensors
PyTorch
English
Chinese
beit
ai-detection
ai-image-detection
deepfake-detection
fake-image-detection
ai-art-detection
stable-diffusion-detection
midjourney-detection
dall-e-detection
image-forensics
digital-art-verification
vit
computer-vision
Eval Results (legacy)
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -76,6 +76,7 @@ def predict(image: Image.Image):
|
|
| 76 |
api_output = {
|
| 77 |
"ai_probability": round(ai_prob, 3),
|
| 78 |
"human_probability": round(human_prob, 3),
|
|
|
|
| 79 |
"top3_sources": top3_sources
|
| 80 |
}
|
| 81 |
|
|
|
|
| 76 |
api_output = {
|
| 77 |
"ai_probability": round(ai_prob, 3),
|
| 78 |
"human_probability": round(human_prob, 3),
|
| 79 |
+
"predicted_source": predicted_source,
|
| 80 |
"top3_sources": top3_sources
|
| 81 |
}
|
| 82 |
|