init
Browse files
app.py
CHANGED
@@ -66,6 +66,12 @@ def load_checkpoint():
|
|
66 |
|
67 |
print(os.listdir(dir + '/checkpoints'))
|
68 |
|
|
|
|
|
|
|
|
|
|
|
|
|
69 |
return dir + '/checkpoints'
|
70 |
|
71 |
|
|
|
66 |
|
67 |
print(os.listdir(dir + '/checkpoints'))
|
68 |
|
69 |
+
dataset_path = huggingface_hub.hf_hub_download(MODEL_REPO,
|
70 |
+
'dataset.zip',
|
71 |
+
force_filename='dataset.zip')
|
72 |
+
print(checkpoint_path)
|
73 |
+
shutil.unpack_archive(dataset_path)
|
74 |
+
|
75 |
return dir + '/checkpoints'
|
76 |
|
77 |
|