Never mind! I just found I used a global variable for the get_country function.
I have the same issue with get_country and get_accuracy.
There was a problem grading your submission. Check stderr for more details.
My lab code is
qyrayhto.z
After checking again with fresh eyes, realized that I had made the same mistake as pointed by Mubsi. I had used word_embeddings in the get_country() function. Now it works.