Update app.py

app.py

@@ -34,7 +34,7 @@ def get_dataloader(start, end, batch_size=8):
     subset = torch.utils.data.Subset(global_data, range(start, end))
     return DataLoader(subset, batch_size=batch_size)
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=20)
 def train_batch(dataloader):
     model.train()
     start_time = time.time()