Spaces:
Runtime error
Runtime error
Commit
·
50cebe2
1
Parent(s):
f0515e4
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ class CPU_Unpickler(pickle.Unpickler):
|
|
9 |
if module == 'torch.storage' and name == '_load_from_bytes':
|
10 |
return lambda b: torch.load(io.BytesIO(b), map_location='cpu')
|
11 |
else:
|
12 |
-
|
13 |
|
14 |
#contents = pickle.load(f) becomes...
|
15 |
#contents = CPU_Unpickler(f).load()
|
|
|
9 |
if module == 'torch.storage' and name == '_load_from_bytes':
|
10 |
return lambda b: torch.load(io.BytesIO(b), map_location='cpu')
|
11 |
else:
|
12 |
+
return super().find_class(module, name)
|
13 |
|
14 |
#contents = pickle.load(f) becomes...
|
15 |
#contents = CPU_Unpickler(f).load()
|