sergeyfeldman commited on
Commit
16683e1
·
verified ·
1 Parent(s): c2ab707

Token usage fix

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -8,9 +8,19 @@ from datetime import datetime
8
 
9
  # set up model
10
  authtoken = os.environ.get("TOKEN")
11
- tokenizer = RobertaTokenizer.from_pretrained("guidecare/feelings_and_issues_large_v2", token=authtoken, use_safetensors=True)
 
 
 
 
 
12
  tokenizer.do_lower_case = True
13
- model = AutoModelForSequenceClassification.from_pretrained("guidecare/feelings_and_issues_large_v2", token=authtoken, use_safetensors=True)
 
 
 
 
 
14
  all_label_names = list(model.config.id2label.values())
15
 
16
  def predict(text):
 
8
 
9
  # set up model
10
  authtoken = os.environ.get("TOKEN")
11
+
12
+ tokenizer = RobertaTokenizer.from_pretrained(
13
+ "guidecare/feelings_and_issues_large_v2",
14
+ use_safetensors=True,
15
+ use_auth_token=authtoken
16
+ )
17
  tokenizer.do_lower_case = True
18
+ model = AutoModelForSequenceClassification.from_pretrained(
19
+ "guidecare/feelings_and_issues_large_v2",
20
+ use_safetensors=True,
21
+ use_auth_token=authtoken
22
+ )
23
+
24
  all_label_names = list(model.config.id2label.values())
25
 
26
  def predict(text):