chansung commited on
Commit
38e3100
Β·
1 Parent(s): 83190db

upload v1664849812 model

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ from huggingface_hub import from_pretrained_keras
8
  PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
10
 
11
- # test_model should be like chansung/test-vit
12
- # test_version
13
- MODEL_CKPT = "test_model@test_version"
14
  MODEL = from_pretrained_keras(MODEL_CKPT)
15
 
16
  RESOLTUION = 224
 
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