Hugging Face's logo

Spaces:
chansung
/
tfx-wandb-test
Sleeping

App Files Community
chansung commited on
Commit
44b42ae
·
1 Parent(s): f666ab2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,8 @@ import tensorflow as tf
8
  from transformers import ViTFeatureExtractor
9
  from huggingface_hub import from_pretrained_keras
10
 
 
 
11
  PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
12
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
13
 
 
8
  from transformers import ViTFeatureExtractor
9
  from huggingface_hub import from_pretrained_keras
10
 
11
+ global MODEL
12
+
13
  PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
14
  feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
15