chrisjay commited on
Commit
7583157
1 Parent(s): 46eef9f

save models and metrics to hub

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. data_mnist +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
40
  MODEL_REPO = 'mnist-adversarial-model'
41
  HF_DATASET ="mnist-adversarial-dataset"
42
  DATASET_REPO_URL = f"https://huggingface.co/datasets/chrisjay/{HF_DATASET}"
43
- MODEL_REPO_URL = f"https://huggingface.co/datasets/chrisjay/{MODEL_REPO}"
44
 
45
  repo = Repository(
46
  local_dir="data_mnist", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
@@ -48,7 +48,7 @@ repo = Repository(
48
  repo.git_pull()
49
 
50
  model_repo = Repository(
51
- local_dir=MODEL_PATH, clone_from=MODEL_REPO_URL, use_auth_token=HF_TOKEN
52
  )
53
  model_repo.git_pull()
54
 
 
40
  MODEL_REPO = 'mnist-adversarial-model'
41
  HF_DATASET ="mnist-adversarial-dataset"
42
  DATASET_REPO_URL = f"https://huggingface.co/datasets/chrisjay/{HF_DATASET}"
43
+ MODEL_REPO_URL = f"https://huggingface.co/model/chrisjay/{MODEL_REPO}"
44
 
45
  repo = Repository(
46
  local_dir="data_mnist", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 
48
  repo.git_pull()
49
 
50
  model_repo = Repository(
51
+ local_dir=MODEL_PATH, clone_from=MODEL_REPO_URL, use_auth_token=HF_TOKEN, repo_type="model"
52
  )
53
  model_repo.git_pull()
54
 
data_mnist CHANGED
@@ -1 +1 @@
1
- Subproject commit 5915a9276e314d92a5b533b5312616b28b9bcee5
 
1
+ Subproject commit fc4a0229b36d92306494d4d1cafb256d1f480046