Can this code still run on the current version of Google Collab? (10/07/2025) I tried downgrading TensorFlow to 2.8.0, but it didn't work. Can you update the code above to support the latest version of Google Collab?