diff --git a/src/remove_ai_watermarks/classify.py b/src/remove_ai_watermarks/classify.py index 6a14ef0..fc07bb6 100644 --- a/src/remove_ai_watermarks/classify.py +++ b/src/remove_ai_watermarks/classify.py @@ -21,7 +21,7 @@ log = logging.getLogger(__name__) CLASSIFY_EXTRA = "'remove-ai-watermarks[classify]'" WEIGHTS_ENV = "RAIW_CLASSIFY_WEIGHTS" WEIGHTS_REPO = "wiltodelta/raiw-photo-classify" -WEIGHTS_REVISION = "ffc46db5135ee3a83f51538f2c8f7483b9b8b40c" +WEIGHTS_REVISION = "4c2763766dd1c8c212d64e01e1cc3f166243c47d" CLIP_FILE = "clip-l-ft.pt" PROBE_FILE = "probe-weights-clip-l-ft.npz" DETECTOR_FILE = "detector.pt"