restest load USE

BiGHeaDMaX · Nov 8, 2023 · 6d478e8 · 6d478e8
1 parent 4c5d503
commit 6d478e8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/app/main.py b/app/main.py
@@ -20,6 +20,9 @@
 # Chargement du modèle de prédiction préentraîné #
 ##################################################
 
+import tensorflow_hub as hub
+embed = hub.load("https://tfhub.dev/google/universal-sentence-encoder/4")
+
 # Contient le modèle KNN et le MultiLabelBinarizer
 # Je passe ici par une fonction, afin de permettre les tests unitaires
 # avec pytest, qui gère mal les imports de fichiers pickle

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,6 @@
 fastapi==0.103.1
 uvicorn==0.23.2
 scikit-learn==1.2.2
-numpy
+numpy
+tensorflow
+tensorflow-hub