Hugging Face's logo

Spaces:
SARVM
/
Deepfake_Detection
Runtime error

App Files Community
S-Mithilesh commited on
Commit
4eed041
·
verified ·
1 Parent(s): aefa610

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ model_spec.loader.exec_module(model_module)
72
 
73
  # Get the actual classes
74
  ViTRefinerConfig = config_module.ViTRefinerConfig
75
- ViTRefinerForImageClassification = model_module.ViTRefinerForImageClassification
76
 
77
  # Register with AutoConfig and AutoModel so from_pretrained works
78
  from transformers import AutoConfig, AutoModelForImageClassification
 
72
 
73
  # Get the actual classes
74
  ViTRefinerConfig = config_module.ViTRefinerConfig
75
+ ViTRefinerForImageClassification = model_module.ViTForImageClassification
76
 
77
  # Register with AutoConfig and AutoModel so from_pretrained works
78
  from transformers import AutoConfig, AutoModelForImageClassification