Update app.py

app.py

@@ -8,8 +8,6 @@ import tensorflow as tf
 from transformers import ViTFeatureExtractor
 from huggingface_hub import from_pretrained_keras
 
-global MODEL
-
 PRETRAIN_CHECKPOINT = "google/vit-base-patch16-224-in21k"
 feature_extractor = ViTFeatureExtractor.from_pretrained(PRETRAIN_CHECKPOINT)
 
@@ -48,6 +46,8 @@ def preprocess_input(image):
     }
 
 def get_predictions(wb_token, image):
+    print(MODEL)
+    
     if MODEL is None:
         wandb.login(key=wb_token)
         path = wandb.use_artifact('tfx-vit-pipeline/final_model:latest', type='model').download()