sergeyfeldman commited on
Commit
431582d
·
1 Parent(s): e0faa7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model = AutoModelForSequenceClassification.from_pretrained("guidecare/feelings_a
11
  all_label_names = list(model.config.id2label.values())
12
 
13
 
14
- def probs(text):
15
  probs = expit(model(**tokenizer([texts], return_tensors="pt", padding=True)).logits.detach().numpy())
16
  return list(zip(all_label_names, probs[0]))
17
 
 
11
  all_label_names = list(model.config.id2label.values())
12
 
13
 
14
+ def predict(text):
15
  probs = expit(model(**tokenizer([texts], return_tensors="pt", padding=True)).logits.detach().numpy())
16
  return list(zip(all_label_names, probs[0]))
17