Update app.py

app.py

@@ -607,7 +607,7 @@ def train_model(data_path, epochs, batch_size, learning_rate, image_size, save_n
         train_dataset = ChartDataset(data_path, image_size=image_size, split="train")
         train_loader = DataLoader(
             train_dataset, batch_size=batch_size, shuffle=True,
-            num_workers=2, pin_memory=True, drop_last=True, collate_fn=collate_fn
+            num_workers=0, pin_memory=True, drop_last=True, collate_fn=collate_fn
         )
         
         # Optimizer