Update app.py

app.py

@@ -108,6 +108,7 @@ def finetune(base_model_path, peptide_length):   #, train_dataset, test_dataset)
     gradient_accumulation_steps=2,
     push_to_hub=True,  #jw 20240918
     hub_token = HF_TOKEN, #jw 20240918
+    dataloader_pin_memory=False， #jw 20241119 true for CPU
     )
 
     # Initialize Trainer