MamaPearl
/

nula-cifar10-robust-v0

Image Classification

computer-vision

adversarial-robustness

Model card Files Files and versions

MamaPearl commited on Apr 7

Commit

c07ea52

·

verified ·

1 Parent(s): 4df515c

Update infer.py

Files changed (1) hide show

infer.py +1 -1

infer.py CHANGED Viewed

@@ -34,7 +34,7 @@ def predict(model, image_source: str, device: str = "cpu") -> dict:
 if __name__ == "__main__":
     import sys
-    source = sys.argv[1] if len(sys.argv) > 1 else "https://upload.wikimedia.org/wikipedia/commons/a/a7/Camponotus_flavomarginatus_ant.jpg"  # an ant for fallback!
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

 if __name__ == "__main__":
     import sys
+    source = sys.argv[1] if len(sys.argv) > 1 else "https://upload.wikimedia.org/wikipedia/commons/a/a7/Camponotus_flavomarginatus_ant.jpg"
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"