Shivam29rathore commited on
Commit
50cebe2
·
1 Parent(s): f0515e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- return super().find_class(module, name)
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()