chansung commited on
Commit
46010ec
β€’
1 Parent(s): 613a9bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -8,8 +8,6 @@ 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
- # 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
 
 
8
  PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
9
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
10
 
 
 
11
  MODEL_CKPT = "chansung/vit-e2e-pipeline-hf-integration@v1664863171"
12
  MODEL = from_pretrained_keras(MODEL_CKPT)
13