Update app.py

app.py

@@ -642,6 +642,7 @@ def train_model(data_path, epochs, batch_size, learning_rate, image_size, save_n
                 
                 epoch_loss += loss.item()
                 current_step += 1
+                print(f"Step {current_step}, loss: {loss.item():.4f}")
             
             avg_loss = epoch_loss / len(train_loader)
             logs.append(f"Epoch {epoch+1}/{epochs}: loss = {avg_loss:.4f}")