Spaces:
Running
Running
Update app.py
Browse files
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)
|