Spaces:

AJain1234
/

Deep_Image_Search_System

Runtime error

AJain1234 commited on Jul 9, 2024

Commit

adac17e

verified ·

1 Parent(s): 26e20fb

Update Experiments/Resnet50_classification.py

Files changed (1) hide show

Experiments/Resnet50_classification.py CHANGED Viewed

@@ -70,7 +70,7 @@ def predict(features_path,image):
       for i in range(len(y_data)):
         train_x.append(x_data[i])
-    features = torch.load(features_path)
     resnet_train_data = []
     for i in range(len(features)):

       for i in range(len(y_data)):
         train_x.append(x_data[i])
+    features = torch.load(features_path , map_location=torch.device('cuda' if torch.cuda.is_available() else 'cpu'))
     resnet_train_data = []
     for i in range(len(features)):