There isn’t much student life happening at US colleges this fall. Not many schools are meeting in person and we have no way of knowing when that will change.
If you study in the US you’ll be expected to return home after graduation. If you’re in one of the top CS programs in Sweden and can do a master’s there, I’d complete their program.