chansung commited on
Commit
e44153e
·
1 Parent(s): d0c9b42

upload 1683870912 model

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -47,8 +47,9 @@ def get_predictions(wb_token, image):
47
  global MODEL
48
 
49
  if MODEL is None:
50
- api = wandb.Api(api_key=wb_token)
51
- path = api.artifact('tfx-vit-pipeline/final_model:1683864255', type='model').download()
 
52
 
53
  tar = tarfile.open(f"{path}/model.tar.gz")
54
  tar.extractall(path=".")
 
47
  global MODEL
48
 
49
  if MODEL is None:
50
+ wandb.login(key=wb_token)
51
+ wandb.init(project="tfx-vit-pipeline", id="avi9pkis", resume=True)
52
+ path = wandb.use_artifact('tfx-vit-pipeline/final_model:1683870912', type='model').download()
53
 
54
  tar = tarfile.open(f"{path}/model.tar.gz")
55
  tar.extractall(path=".")