Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,6 @@ repo_id = "ChaolongYang/KDTalker"
|
|
11 |
local_dir = "./downloaded_repo"
|
12 |
snapshot_download(repo_id=repo_id, local_dir=local_dir)
|
13 |
print("Download complete! Repository saved at:", local_dir)
|
14 |
-
cache_dir = hf_hub_cache_dir()
|
15 |
-
print("Hugging Face cache directory:", cache_dir)
|
16 |
|
17 |
|
18 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
11 |
local_dir = "./downloaded_repo"
|
12 |
snapshot_download(repo_id=repo_id, local_dir=local_dir)
|
13 |
print("Download complete! Repository saved at:", local_dir)
|
|
|
|
|
14 |
|
15 |
|
16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|