ELMo_hub.py

import tensorflow_hub as hub
import tensorflow as tf

words_to_embed = ["dog", "cat", "sloth"] # <-- it's a list now, and the name changed

elmo = hub.Module("https://tfhub.dev/google/elmo/2")
embedding_tensor = elmo(words_to_embed, as_dict=True)["word_emb"] # <-- passing in a list instead of [word]

with tf.Session() as sess:
  sess.run(tf.global_variables_initializer())
  embedding = sess.run(embedding_tensor)
  print(embedding.shape)