Spaces:

offry
/

Arcaid

Runtime error

offry commited on May 5, 2023

Commit

10fb420

1 Parent(s): 6e22eb4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -344,7 +344,7 @@ with open('query_images_paths.pkl', 'rb') as fp:
     query_images_paths = pickle.load(fp)
 with open('features.pkl', 'rb') as fp:
-    features = pickle.load(fp, map_location=torch.device('cpu'))

     query_images_paths = pickle.load(fp)
 with open('features.pkl', 'rb') as fp:
+    features = pickle.load(fp)