TKM03 commited on
Commit
52035bd
·
verified ·
1 Parent(s): 6098ec9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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}")