osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero commited on Feb 21, 2022

Commit

1931f62

•

1 Parent(s): 56f15e5

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -5

pipeline.py CHANGED Viewed

@@ -12,11 +12,8 @@ class PreTrainedPipeline():
         """
         Initialize model
         """
-        try:
-            self.model = BigGAN.from_pretrained(path)
-        except (IOError, OSError):
-            nltk.download('wordnet')
-            self.model = BigGAN.from_pretrained(path)
         self.truncation = 0.1
     def __call__(self, inputs: str):

         """
         Initialize model
         """
+        nltk.download('wordnet')
+        self.model = BigGAN.from_pretrained(path)
         self.truncation = 0.1
     def __call__(self, inputs: str):