Update pipeline.py

pipeline.py

@@ -23,10 +23,9 @@ class PreTrainedPipeline():
             A :obj:`PIL.Image`. The raw image representation as PIL.
         """
         class_vector = one_hot_from_names([inputs], batch_size=1)
-        if class_vector == None:
+        if type(class_vector) == type(None):
             raise ValueError("Input is not in ImageNet")
 
-
         noise_vector = truncated_noise_sample(truncation=self.truncation, batch_size=1)
 
         noise_vector = torch.from_numpy(noise_vector)