Update app.py

app.py

@@ -104,7 +104,7 @@ if (runModel=='1'):
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
-        num_train_epochs=10,
+        num_train_epochs=8,
         per_device_train_batch_size=2,
         per_device_eval_batch_size=2,
         warmup_steps=500,
@@ -172,8 +172,8 @@ if (runModel=='1'):
     train_dataloader = DataLoader(
         train_dataset, 
         batch_size=10, 
-        shuffle=True, 
-        num_workers=4  # Increase workers for faster data loading
+        shuffle=True
+        #num_workers=4  # Increase workers for faster data loading
     )
     
     evaluate_and_report_errors(model,train_dataloader, tokenizer)