init

app.py

@@ -65,7 +65,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             model_inputs["labels"] = labels["input_ids"]
             return model_inputs
     
-        tokenized_datasets = dataset.map(tokenize_function, batched=True, batch_size=6)
+        tokenized_datasets = dataset.map(tokenize_function, batched=True, batch_size=32)
         data_collator = DataCollatorForSeq2Seq(tokenizer, model=model)
     
         # Set training arguments