Spaces:
Sleeping
Sleeping
Commit ·
fb8a7a1
1
Parent(s): 58e7083
Update app.py
Browse filesdisable weights-only mode
app.py
CHANGED
|
@@ -46,7 +46,7 @@ class Vocab:
|
|
| 46 |
if not os.path.exists(CACHE_FILE):
|
| 47 |
raise FileNotFoundError(f"Cache file {CACHE_FILE} not found. Please upload ubuntu_data_cache.pt")
|
| 48 |
|
| 49 |
-
cache = torch.load(CACHE_FILE, map_location="cpu")
|
| 50 |
train_pairs = cache["train"]
|
| 51 |
test_pairs = cache["test"]
|
| 52 |
vocab = cache["vocab"]
|
|
|
|
| 46 |
if not os.path.exists(CACHE_FILE):
|
| 47 |
raise FileNotFoundError(f"Cache file {CACHE_FILE} not found. Please upload ubuntu_data_cache.pt")
|
| 48 |
|
| 49 |
+
cache = torch.load(CACHE_FILE, map_location="cpu", weights_only=False)
|
| 50 |
train_pairs = cache["train"]
|
| 51 |
test_pairs = cache["test"]
|
| 52 |
vocab = cache["vocab"]
|