Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -27,10 +27,11 @@ hyper_params = {
|
|
27 |
"label2id": label2id,
|
28 |
}
|
29 |
|
30 |
-
|
31 |
|
32 |
# Load the processor manually
|
33 |
-
vit_img_processor = AutoImageProcessor.from_pretrained(
|
|
|
34 |
|
35 |
class DeepFakeModel(L.LightningModule):
|
36 |
def __init__(self, hyperparams: dict):
|
|
|
27 |
"label2id": label2id,
|
28 |
}
|
29 |
|
30 |
+
|
31 |
|
32 |
# Load the processor manually
|
33 |
+
vit_img_processor = AutoImageProcessor.from_pretrained('microsoft/swin-small-patch4-window7-224')
|
34 |
+
|
35 |
|
36 |
class DeepFakeModel(L.LightningModule):
|
37 |
def __init__(self, hyperparams: dict):
|