Update app.py

app.py

@@ -31,8 +31,7 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
     tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
     
-    #batch_size = 16
-    batch_size = 8
+    batch_size = 16
     args = TrainingArguments(
         "test-glue",
         eval_strategy = "epoch",
@@ -47,23 +46,6 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
         metric_for_best_model="accuracy",
         report_to="none"
         )
-    """
-    batch_size = 1024
-    args = TrainingArguments(
-        "test-glue",
-        eval_strategy = "epoch",
-        learning_rate=0.0007199555649276667,
-        seed=33,
-        lr_scheduler_type="linear",
-        per_device_train_batch_size=batch_size,
-        per_device_eval_batch_size=batch_size,
-        num_train_epochs=7,
-        weight_decay=0.01,
-        load_best_model_at_end=False,
-        metric_for_best_model="accuracy",
-        report_to="none"
-        )
-    """
     trainer = Trainer(
         model,
         args,