Update seefood_classifier_import.py
Browse files
seefood_classifier_import.py
CHANGED
@@ -18,7 +18,7 @@ def install_packages(packages_path: Path):
|
|
18 |
|
19 |
def get_seefood_classifier():
|
20 |
model_repo = 'ronig/seefood'
|
21 |
-
local_repo_path = Path(snapshot_download(model_repo,
|
22 |
subprocess.run(["pip", "install", "-r", str(local_repo_path / 'requirements.txt')])
|
23 |
install_packages(local_repo_path / 'build' / 'dist')
|
24 |
sys.path.append(str(local_repo_path))
|
|
|
18 |
|
19 |
def get_seefood_classifier():
|
20 |
model_repo = 'ronig/seefood'
|
21 |
+
local_repo_path = Path(snapshot_download(model_repo, token=os.environ['TOKEN']))
|
22 |
subprocess.run(["pip", "install", "-r", str(local_repo_path / 'requirements.txt')])
|
23 |
install_packages(local_repo_path / 'build' / 'dist')
|
24 |
sys.path.append(str(local_repo_path))
|