Fix Sentence Transformers import location

app.py

@@ -3,7 +3,6 @@ from huggingface_hub import hf_hub_download
 import json
 import tensorflow as tf
 import numpy as np
-from sentence_transformers import SentenceTransformer
 
 
 # Load models
@@ -17,7 +16,8 @@ nn_model = tf.keras.models.load_model(nn_model_path)
 
 llm_model_path = hf_hub_download(repo_id="tbitai/gisty-enron1-spam", filename="gisty-enron1-spam.keras")
 llm_model = tf.keras.models.load_model(llm_model_path)
-# SentenceTransformer should be called after Keras models, in order to prevent it from setting Keras to legacy.
+# Sentence Transformers should be imported after Keras models, in order to prevent it from setting Keras to legacy.
+from sentence_transformers import SentenceTransformer
 st_model = SentenceTransformer("avsolatorio/GIST-large-Embedding-v0")
 
 # Utils for Bayes