Spaces:
Runtime error
Runtime error
Greg Thompson
commited on
Commit
•
edd10ba
1
Parent(s):
9f56df0
Move model to global namespace
Browse files
mathtext_fastapi/intent_classification.py
CHANGED
@@ -42,7 +42,10 @@ def retrieve_intent_classification_model():
|
|
42 |
|
43 |
|
44 |
encoder = SentenceTransformer('all-MiniLM-L6-v2')
|
45 |
-
model = retrieve_intent_classification_model()
|
|
|
|
|
|
|
46 |
|
47 |
def predict_message_intent(message):
|
48 |
tokenized_utterance = np.array([list(encoder.encode(message))])
|
|
|
42 |
|
43 |
|
44 |
encoder = SentenceTransformer('all-MiniLM-L6-v2')
|
45 |
+
# model = retrieve_intent_classification_model()
|
46 |
+
DATA_DIR = Path(__file__).parent.parent / "mathtext_fastapi" / "data" / "intent_classification_model.joblib"
|
47 |
+
model = load(DATA_DIR)
|
48 |
+
|
49 |
|
50 |
def predict_message_intent(message):
|
51 |
tokenized_utterance = np.array([list(encoder.encode(message))])
|