Rename app.py to app.pyhttps://huggingface.co/spaces/LPX55/Deepfake-Detection-Playground-Dev/settings
Browse files
app.py → app.pyhttps:/huggingface.co/spaces/LPX55/Deepfake-Detection-Playground-Dev/settings
RENAMED
|
@@ -64,23 +64,21 @@ class NumpyEncoder(json.JSONEncoder):
|
|
| 64 |
# Ensure using GPU if available
|
| 65 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 66 |
|
| 67 |
-
# Model paths and class names (copied from app_mcp.py)
|
| 68 |
MODEL_PATHS = {
|
| 69 |
"model_1": "LPX55/detection-model-1-ONNX",
|
| 70 |
"model_2": "LPX55/detection-model-2-ONNX",
|
| 71 |
"model_3": "LPX55/detection-model-3-ONNX",
|
| 72 |
-
"model_4": "cmckinle/sdxl-flux-detector_v1.1",
|
| 73 |
"model_5": "LPX55/detection-model-5-ONNX",
|
| 74 |
"model_6": "LPX55/detection-model-6-ONNX",
|
| 75 |
"model_7": "LPX55/detection-model-7-ONNX",
|
| 76 |
"model_8": "aiwithoutborders-xyz/CommunityForensics-DeepfakeDet-ViT"
|
| 77 |
}
|
| 78 |
|
|
|
|
| 79 |
CLASS_NAMES = {
|
| 80 |
"model_1": ['artificial', 'real'],
|
| 81 |
"model_2": ['AI Image', 'Real Image'],
|
| 82 |
"model_3": ['artificial', 'human'],
|
| 83 |
-
"model_4": ['AI', 'Real'],
|
| 84 |
"model_5": ['Realism', 'Deepfake'],
|
| 85 |
"model_6": ['ai_gen', 'human'],
|
| 86 |
"model_7": ['Fake', 'Real'],
|
|
|
|
| 64 |
# Ensure using GPU if available
|
| 65 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 66 |
|
|
|
|
| 67 |
MODEL_PATHS = {
|
| 68 |
"model_1": "LPX55/detection-model-1-ONNX",
|
| 69 |
"model_2": "LPX55/detection-model-2-ONNX",
|
| 70 |
"model_3": "LPX55/detection-model-3-ONNX",
|
|
|
|
| 71 |
"model_5": "LPX55/detection-model-5-ONNX",
|
| 72 |
"model_6": "LPX55/detection-model-6-ONNX",
|
| 73 |
"model_7": "LPX55/detection-model-7-ONNX",
|
| 74 |
"model_8": "aiwithoutborders-xyz/CommunityForensics-DeepfakeDet-ViT"
|
| 75 |
}
|
| 76 |
|
| 77 |
+
# CLASS_NAMES (model_4 removed)
|
| 78 |
CLASS_NAMES = {
|
| 79 |
"model_1": ['artificial', 'real'],
|
| 80 |
"model_2": ['AI Image', 'Real Image'],
|
| 81 |
"model_3": ['artificial', 'human'],
|
|
|
|
| 82 |
"model_5": ['Realism', 'Deepfake'],
|
| 83 |
"model_6": ['ai_gen', 'human'],
|
| 84 |
"model_7": ['Fake', 'Real'],
|