Put specifying args.device in main file instead of inside function 9b4dec3 verified nikigoli commited on Jul 6
Changed device to global variable in gradio (ie gr.State instance) 31bd1d5 verified nikigoli commited on Jul 6