specific exception handle for reading sscd model's file not found exc…

…eption
meedan · Jan 29, 2024 · 80ae3fd · 80ae3fd
1 parent 1a7b01b
commit 80ae3fd
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:
+        except FileNotFoundError:
             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: