sergeyfeldman commited on
Commit
c2a87e9
·
verified ·
1 Parent(s): b749f9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ from datetime import datetime
8
 
9
  # set up model
10
  authtoken = os.environ.get("TOKEN") or True
11
- tokenizer = RobertaTokenizer.from_pretrained("guidecare/feelings_and_issues_large_v2", use_auth_token=authtoken)
12
  tokenizer.do_lower_case = True
13
- model = AutoModelForSequenceClassification.from_pretrained("guidecare/feelings_and_issues_large_v2", use_auth_token=authtoken)
14
  all_label_names = list(model.config.id2label.values())
15
 
16
 
 
8
 
9
  # set up model
10
  authtoken = os.environ.get("TOKEN") or True
11
+ tokenizer = RobertaTokenizer.from_pretrained("guidecare/feelings_and_issues_large_v2", use_auth_token=authtoken, use_safetensors=True)
12
  tokenizer.do_lower_case = True
13
+ model = AutoModelForSequenceClassification.from_pretrained("guidecare/feelings_and_issues_large_v2", use_auth_token=authtoken, use_safetensors=True)
14
  all_label_names = list(model.config.id2label.values())
15
 
16