Spaces:

marco-pcg
/

Neural-Network-Types

Sleeping

marco-pcg commited on Oct 2

Commit

d6da7be

1 Parent(s): 407f1ce

fix: import huggingface_hub

Files changed (1) hide show

inference/autoencoder_leaks.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch.nn as nn
 MODEL_DIR = Path("train/models")
 model_path = hf_hub_download(
     repo_id="marco-pcg/Neural-Network-Types-models",

 MODEL_DIR = Path("train/models")
+from huggingface_hub import hf_hub_download
 model_path = hf_hub_download(
     repo_id="marco-pcg/Neural-Network-Types-models",