Running KGAT in Google Collab

[mabner1996]

Hello, is it possible to run this repository in google collab?
I tried to upload this repo to google drive, mount it via google collab and run the amazon example :
(python Main.py --model_type kgat --alg_type bi --dataset amazon-book --regs [1e-5,1e-5] --layer_size [64,32,16] --embed_size 64 --lr 0.0001 --epoch 1000 --verbose 50 --save_flag 1 --pretrain -1 --batch_size 1024 --node_dropout [0.1] --mess_dropout [0.1,0.1,0.1] --use_att True --use_kge True)

However, I get an error message saying it cannot find train.txt in the directory. I put the screenshot of the command I input in collab and error message

I hope you can suggest other way to solve this/ running it on collab, thanks!