# Test notebook import os from google.colab import drive # Mount Google Drive drive.mount('/content/gdrive') # Run the main function main()