Spaces:

WesScivetti
/

SNACS_English_Demo

Running on Zero

WesScivetti commited on Jun 10

Commit

c5bb23a

1 Parent(s): 67c2a03

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -195,7 +195,7 @@ def classify_tokens(text):
               'p.Whole-p.Circumstance': '#c70411',
               'p.Purpose-p.Goal': '#f2f199'}
-    token_classifier = pipeline("token-classification", model="WesScivetti/SNACS_English",
                                 aggregation_strategy="simple")
     results = token_classifier(text)

               'p.Whole-p.Circumstance': '#c70411',
               'p.Purpose-p.Goal': '#f2f199'}
+    token_classifier = pipeline("token-classification", model="WesScivetti/SNACS_Multilingual",
                                 aggregation_strategy="simple")
     results = token_classifier(text)