Update app.py

app.py

@@ -89,7 +89,7 @@ def train_model():
 
     print("🔁 Training started...")
     model.train()
-    for epoch in range(2):  # Train for 5 epochs
+    for epoch in range(1):  # Train for 5 epochs
         running_loss = 0.0
         for images, labels in train_loader:
             optimizer.zero_grad()