Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,8 +9,8 @@ logging.basicConfig(level=logging.INFO)
|
|
9 |
logger = logging.getLogger(__name__)
|
10 |
|
11 |
# Load the model and processor from Hugging Face
|
12 |
-
model = ViTForImageClassification.from_pretrained("prithivMLmods/Deep-Fake-Detector-Model")
|
13 |
-
processor = ViTImageProcessor.from_pretrained("prithivMLmods/Deep-Fake-Detector-Model")
|
14 |
|
15 |
# Log model configuration to verify label mapping
|
16 |
logger.info(f"Model label mapping: {model.config.id2label}")
|
|
|
9 |
logger = logging.getLogger(__name__)
|
10 |
|
11 |
# Load the model and processor from Hugging Face
|
12 |
+
model = ViTForImageClassification.from_pretrained("prithivMLmods/Deep-Fake-Detector-v2-Model")
|
13 |
+
processor = ViTImageProcessor.from_pretrained("prithivMLmods/Deep-Fake-Detector-v2-Model")
|
14 |
|
15 |
# Log model configuration to verify label mapping
|
16 |
logger.info(f"Model label mapping: {model.config.id2label}")
|