Changing FIleNotFoundError to ValueError

meedan · Jan 29, 2024 · a012808 · a012808
1 parent 80ae3fd
commit a012808
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/model/image_sscd.py b/lib/model/image_sscd.py
@@ -19,7 +19,7 @@ def __init__(self):
         #FIXME: Load from a Meedan S3 bucket
         try:
             self.model = torch.jit.load("sscd_disc_mixup.torchscript.pt")
-        except FileNotFoundError:
+        except ValueError:
             logger.info("Downloading SSCD model...")
             m=urllib.request.urlopen("https://dl.fbaipublicfiles.com/sscd-copy-detection/sscd_disc_mixup.torchscript.pt").read()
             with open("sscd_disc_mixup.torchscript.pt","wb") as fh: