Spaces:
Sleeping
Sleeping
upload 1683870912 model
Browse files
app.py
CHANGED
@@ -47,8 +47,9 @@ def get_predictions(wb_token, image):
|
|
47 |
global MODEL
|
48 |
|
49 |
if MODEL is None:
|
50 |
-
|
51 |
-
|
|
|
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=".")
|