ksatzke commited on
Commit
117043a
·
verified ·
1 Parent(s): 452ea23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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,