Spaces:

Oumar199
/

Fake-Real-Face-Detection

Build error

= commited on May 27, 2023

Commit

7b67be2

1 Parent(s): 110ba75

adding checkpoint

Files changed (1) hide show

app.py CHANGED Viewed

@@ -52,7 +52,7 @@ def get_model():
     # recuperate the model
     model = ViTForImageClassification.from_pretrained(
-        'data/checkpoints/model_2yW4AcqNIb6zLKNIb6zLK',
         num_labels = len(characs['ids']),
         id2label = {name: key for key, name in characs['ids'].items()},
         label2id = characs['ids']

     # recuperate the model
     model = ViTForImageClassification.from_pretrained(
+        'data/checkpoints/model_2yW4AcqNIb6zLKNIb6zLK/checkpoint-1500',
         num_labels = len(characs['ids']),
         id2label = {name: key for key, name in characs['ids'].items()},
         label2id = characs['ids']