pujanpaudel commited on
Commit
dd90bd4
·
verified ·
1 Parent(s): 79712c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -27,10 +27,11 @@ hyper_params = {
27
  "label2id": label2id,
28
  }
29
 
30
- preprocessor_config_path = "preprocessor_config.json"
31
 
32
  # Load the processor manually
33
- vit_img_processor = AutoImageProcessor.from_pretrained(preprocessor_config_path)
 
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):