abdulmatinomotoso commited on
Commit
2090148
1 Parent(s): b73ef4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ def get_probs(logits, threshold=0.5):
21
  def multi_label_emotions(text):
22
  inputs = tokenizer(text, return_tensors="pt", truncation=True)
23
  model.to(device)
 
24
  with torch.no_grad():
25
  logits = model(**inputs).logits
26
  #probs = logits.int().numpy()[0]
 
21
  def multi_label_emotions(text):
22
  inputs = tokenizer(text, return_tensors="pt", truncation=True)
23
  model.to(device)
24
+
25
  with torch.no_grad():
26
  logits = model(**inputs).logits
27
  #probs = logits.int().numpy()[0]