【Python】関数内で外部モジュールを利用したい

やりたいこと

自然言語処理とプログラミングを学習中の初心者です．
可能性の話として可能かということとどのように書いたら可能かを聞きたいです．
やりたいこととしては以下です．

• gensimのword2vec.similalityを作成した関数内で取り扱いたい

実行環境は以下です．
Macboookpro(2017)OS:11.6.7
Python:2.7

ソースコード（仮組）

python

# -*- coding: utf-8 -*-import gensim from gensim.models import word2vec def math(a, b): print(a, "+", b, "=", a+b) def cntS(t, d, theta): s = 0 #ここが動くようにしたい for x in d: if model.similarity(t, x) >= theta: s += 1 return s model_dir = './entity_vector/entity_vector.model.bin'model = KeyedVectors.load(model_dir, binary=True) aray = ["米", "麦", "小麦", "豆腐"]s = cntS("雑穀", aray, 0, 4)print(s)math(1, 2)

よろしくお願いします．