Update app.py

app.py

@@ -91,7 +91,8 @@ def train_step(file=None, start_idx=0):
             print("Exceeded GPU quota.")
             if not os.path.exists('./checkpoint'):
                 os.makedirs('./checkpoint')
-                torch.save(model.state_dict(), "./checkpoint/model.pt")
+            print('Save checkpoint')
+            torch.save(model.state_dict(), "./checkpoint/model.pt")
             return start_idx  # Trả về start_idx để lưu lại vị trí
 
         start_idx = end_idx