chansung commited on
Commit
613a9bc
Β·
1 Parent(s): 38e3100

upload v1664863171 model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
10
 
11
  # chansung/vit-e2e-pipeline-hf-integration should be like chansung/test-vit
12
- # v1664849812
13
- MODEL_CKPT = "chansung/vit-e2e-pipeline-hf-integration@v1664849812"
14
  MODEL = from_pretrained_keras(MODEL_CKPT)
15
 
16
  RESOLTUION = 224
 
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
10
 
11
  # chansung/vit-e2e-pipeline-hf-integration should be like chansung/test-vit
12
+ # v1664863171
13
+ MODEL_CKPT = "chansung/vit-e2e-pipeline-hf-integration@v1664863171"
14
  MODEL = from_pretrained_keras(MODEL_CKPT)
15
 
16
  RESOLTUION = 224