Spaces:

SkalskiP
/

MetaCLIP

Sleeping

SkalskiP commited on Oct 26, 2023

Commit

fcb4afd

•

1 Parent(s): 2bdfd2c

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def classify_image(input_image) -> str:
         padding=True).to(device)
     outputs = model(**inputs)
     probs = outputs.logits_per_image.softmax(dim=1)
-    class_index = np.argmax(probs.detach().numpy())
     return imagenet_classes[class_index]

         padding=True).to(device)
     outputs = model(**inputs)
     probs = outputs.logits_per_image.softmax(dim=1)
+    class_index = np.argmax(probs.detach().cpu().numpy())
     return imagenet_classes[class_index]