Update app.py

app.py

@@ -47,9 +47,8 @@ def get_predictions(wb_token, image):
     global MODEL
     
     if MODEL is None:
-        wandb.login(key=wb_token)
-        wandb.init(project="tfx-vit-pipeline")
-        path = wandb.use_artifact('tfx-vit-pipeline/final_model:1683864255', type='model').download()
+        api = wandb.Api(api_key=wb_token)
+        path = api.artifact('tfx-vit-pipeline/final_model:1683864255', type='model').download()
 
         tar = tarfile.open(f"{path}/model.tar.gz")
         tar.extractall(path=".")