Merge pull request #51 from davidberenstein1957/fix/zero-shot-ort-error

Update classy_spacy.py

classy_classification/classifiers/classy_spacy.py

@@ -171,7 +171,7 @@ def set_classification_model(self, model: str = None, device: str = None):
             self.pipeline = pipeline(
                 "zero-shot-classification", model=model, device=self.device, top_k=None, accelerator="ort"
             )
-        except ImportError:
+        except Exception:
             from transformers import pipeline
 
             if self.device in ["gpu", "cuda", 0]:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "classy-classification"
-version = "1.0.0"
+version = "1.0.1"
 description = "Have you every struggled with needing a Spacy TextCategorizer but didn't have the time to train one from scratch? Classy Classification is the way to go!"
 authors = ["David Berenstein <[email protected]>"]
 license = "MIT"
@@ -32,6 +32,7 @@ spacy = {extras = ["transformers"], version = "^3.0"}
 sentence-transformers = "^2.0"
 scikit-learn = "^1.0"
 pandas = ">=1,<2"
+numpy = "<2"
 transformers = {extras = ["torch"], version = ">4.20,<5"}
 terminado = "<0.18"