Update app.py

app.py

@@ -19,7 +19,7 @@ big = True if device == torch.device('cpu') else False
 
 # Parameters
 IMG_SIZE = 1024 if big else 256
-BATCH_SIZE = 16 if big else 4
+BATCH_SIZE = 1 if big else 4
 EPOCHS = 12
 LR = 0.0002
 dataset_id = "K00B404/pix2pix_flux_set"