Update universal_sentence_encoder.py

textattack/constraints/semantics/sentence_encoders/universal_sentence_encoder/universal_sentence_encoder.py

@@ -18,25 +18,16 @@ def __init__(self, threshold=0.8, large=False, metric="angular", **kwargs):
  super().__init__(threshold=threshold, metric=metric, **kwargs)
  if large:
  tfhub_url = "https://tfhub.dev/google/universal-sentence-encoder-large/5"
- mirror_tfhub_url = "https://hub.tensorflow.google.cn/google/universal-sentence-encoder-large/5"
  else:
  tfhub_url = "https://tfhub.dev/google/universal-sentence-encoder/4"
- mirror_tfhub_url = (
- "https://hub.tensorflow.google.cn/google/universal-sentence-encoder/4"
- )
 
  self._tfhub_url = tfhub_url
- self.mirror_tfhub_url = mirror_tfhub_url
  # Lazily load the model
  self.model = None
 
  def encode(self, sentences):
  if not self.model:
- try:
- self.model = hub.load(self._tfhub_url)
- except Exception as e:
- print('Error loading model from tfhub, trying mirror url')
- self.model = hub.load(self.mirror_tfhub_url)
+ self.model = hub.load(self._tfhub_url)
  return self.model(sentences).numpy()
 
  def __getstate__(self):
@@ -46,8 +37,5 @@ def __getstate__(self):
 
  def __setstate__(self, state):
  self.__dict__ = state
- try:
- self.model = hub.load(self._tfhub_url)
- except Exception as e:
- print('Error loading model from tfhub, trying mirror url')
- self.model = hub.load(self.mirror_tfhub_url)
+ self.model = hub.load(self._tfhub_url)
+