vnanhtuan commited on
Commit
427d67b
·
verified ·
1 Parent(s): 5a4357c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  def preprocess_function(examples):
13
  return tokenizer(examples["comment"], truncation=True, padding=True)
14
 
 
 
15
  def compute_metrics(eval_pred):
16
  logits, labels = eval_pred
17
  predictions = torch.argmax(torch.tensor(logits), dim=-1)
 
12
  def preprocess_function(examples):
13
  return tokenizer(examples["comment"], truncation=True, padding=True)
14
 
15
+
16
+
17
  def compute_metrics(eval_pred):
18
  logits, labels = eval_pred
19
  predictions = torch.argmax(torch.tensor(logits), dim=-1)