vives commited on
Commit
8792615
1 Parent(s): 0373598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -37,9 +37,6 @@ with open(kp_dict_checkpoint,'rb') as handle:
37
  kp_dict = pickle.load(handle)
38
  keys = list(kp_dict.keys())
39
 
40
- for key in kp_dict.keys():
41
- kp_dict[key] = kp_dict[key].detach().numpy()
42
-
43
  #load cosine distances of kp dict
44
  with open(kp_cosine_checkpoint,'rb') as handle:
45
  cosine_kp = pickle.load(handle)
 
37
  kp_dict = pickle.load(handle)
38
  keys = list(kp_dict.keys())
39
 
 
 
 
40
  #load cosine distances of kp dict
41
  with open(kp_cosine_checkpoint,'rb') as handle:
42
  cosine_kp = pickle.load(handle)