Update README.md

README.md

@@ -72,7 +72,7 @@ def forward_text_with_templates(text, templates=TEMPLATES, normalize=True):
 def classifier_zero_shot_with_pil(img, classes):
 	classifiers  = torch.cat([forward_text_with_templates(c).reshape([1, -1]) for c in classes], axis=0)
 	img_features = forward_image(img)
-	scores       = torch.matmul(img_features, classifiers.T)[0].cpu().numpy()
+	scores       = torch.matmul(img_features, classifiers.T)[0].detach().cpu().numpy()
 	out          = classes[np.argmax(scores)]
 	return out
 ###############################################################