Update app.py

app.py

@@ -65,7 +65,7 @@ for cls in os.listdir(dataset_folder):
 # 2. Load Data
 # -------------------------------
 transform = transforms.Compose([
-    transforms.Resize((64, 64)),
+    transforms.Resize((128, 128)),
     transforms.ToTensor(),
     transforms.Normalize((0.5,), (0.5,))
 ])
@@ -110,7 +110,7 @@ optimizer = optim.Adam(model.parameters(), lr=0.001)
 # 4. Train Model
 # -------------------------------
 epoch_logs = []
-num_epochs = 10
+num_epochs = 15
 
 for epoch in range(num_epochs):
     model.train()