Spaces:
Runtime error
Runtime error
upload v1664863171 model
Browse files
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 |
-
#
|
13 |
-
MODEL_CKPT = "chansung/vit-e2e-pipeline-hf-integration@
|
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
|