Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def compute_metrics(eval_pred):
|
|
30 |
def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric):
|
31 |
tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
|
32 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
33 |
-
|
34 |
batch_size = 16
|
35 |
args = TrainingArguments(
|
36 |
"test-glue",
|
@@ -62,6 +62,7 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
|
|
62 |
metric_for_best_model="accuracy",
|
63 |
report_to="none"
|
64 |
)
|
|
|
65 |
trainer = Trainer(
|
66 |
model,
|
67 |
args,
|
|
|
30 |
def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric):
|
31 |
tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
|
32 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
33 |
+
|
34 |
batch_size = 16
|
35 |
args = TrainingArguments(
|
36 |
"test-glue",
|
|
|
62 |
metric_for_best_model="accuracy",
|
63 |
report_to="none"
|
64 |
)
|
65 |
+
"""
|
66 |
trainer = Trainer(
|
67 |
model,
|
68 |
args,
|