Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,9 @@ MODEL_MAX_LENGTH = 512
|
|
36 |
AUTH_TOKEN = os.environ.get("TOKEN") or True
|
37 |
# PoS
|
38 |
pos_tokenizer = AutoTokenizer.from_pretrained(
|
39 |
-
MODEL_PATHS["
|
|
|
|
|
40 |
)
|
41 |
pos_model = AutoModelForTokenClassification.from_pretrained(
|
42 |
MODEL_PATHS["POS"], use_auth_token=AUTH_TOKEN, revision="11437fc1ba7ecd7df9d292d5f91f36fe55cdae3d",
|
|
|
36 |
AUTH_TOKEN = os.environ.get("TOKEN") or True
|
37 |
# PoS
|
38 |
pos_tokenizer = AutoTokenizer.from_pretrained(
|
39 |
+
MODEL_PATHS["DEPENDENCY"],
|
40 |
+
model_max_length=MODEL_MAX_LENGTH,
|
41 |
+
use_auth_token=AUTH_TOKEN,
|
42 |
)
|
43 |
pos_model = AutoModelForTokenClassification.from_pretrained(
|
44 |
MODEL_PATHS["POS"], use_auth_token=AUTH_TOKEN, revision="11437fc1ba7ecd7df9d292d5f91f36fe55cdae3d",
|