diff --git a/gate_stance.py b/gate_stance.py index 25cc2fa..3a16356 100644 --- a/gate_stance.py +++ b/gate_stance.py @@ -32,7 +32,7 @@ def run(doc, **kwargs): global classifier if classifier is None: - classifier = StanceClassifier("lr") + classifier = StanceClassifier(kwargs.get("model", "ens")) text = doc.text