LPX55 commited on
Commit
99b6d79
·
verified ·
1 Parent(s): 9bb3ed3

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'],