bowphs commited on
Commit
7b674b2
1 Parent(s): d66b3ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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["POS"], model_max_length=MODEL_MAX_LENGTH, use_auth_token=AUTH_TOKEN, revision="11437fc1ba7ecd7df9d292d5f91f36fe55cdae3d", use_fast=False,
 
 
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",