chansung
/

tfx-vit-pipeline

Model card Files Files and versions Community

chansung commited on May 12, 2023

Commit

c44275a

·

1 Parent(s): 237e09c

upload 1683859487 model

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def get_predictions(wb_token, image):
     if MODEL is None:
         wandb.login(key=wb_token)
         wandb.init(project="tfx-vit-pipeline")
-        path = wandb.use_artifact('tfx-vit-pipeline/final_model:1683855462', type='model').download()
         tar = tarfile.open(f"{path}/model.tar.gz")
         tar.extractall(path=".")

     if MODEL is None:
         wandb.login(key=wb_token)
         wandb.init(project="tfx-vit-pipeline")
+        path = wandb.use_artifact('tfx-vit-pipeline/final_model:1683859487', type='model').download()
         tar = tarfile.open(f"{path}/model.tar.gz")
         tar.extractall(path=".")