Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|