init

app.py

@@ -66,6 +66,12 @@ def load_checkpoint():
 
     print(os.listdir(dir + '/checkpoints'))
 
+    dataset_path = huggingface_hub.hf_hub_download(MODEL_REPO,
+                                                      'dataset.zip',
+                                                      force_filename='dataset.zip')
+    print(checkpoint_path)
+    shutil.unpack_archive(dataset_path)
+
     return dir + '/checkpoints'