Leon Sick
commited on
Commit
·
af3c6d0
1
Parent(s):
1045df5
start of space
Browse files
model.py
CHANGED
|
@@ -96,6 +96,7 @@ def load_model(score_threshold: float) -> VisualizationDemo:
|
|
| 96 |
model_dir.mkdir(exist_ok=True)
|
| 97 |
#weight_path = model_dir / WEIGHT_URL.split('/')[-1]
|
| 98 |
weight_path = model_dir / ".cache" / "huggingface" / "download"/ model_filename
|
|
|
|
| 99 |
print("***weight_path***", weight_path)
|
| 100 |
print("torch.cuda.is_available()", torch.cuda.is_available())
|
| 101 |
|
|
|
|
| 96 |
model_dir.mkdir(exist_ok=True)
|
| 97 |
#weight_path = model_dir / WEIGHT_URL.split('/')[-1]
|
| 98 |
weight_path = model_dir / ".cache" / "huggingface" / "download"/ model_filename
|
| 99 |
+
weight_path = model_dir / model_filename
|
| 100 |
print("***weight_path***", weight_path)
|
| 101 |
print("torch.cuda.is_available()", torch.cuda.is_available())
|
| 102 |
|