diff --git a/scripts/install b/scripts/install
index a7e11e1..7b1774c 100755
--- a/scripts/install
+++ b/scripts/install
@@ -134,7 +134,8 @@ pushd "$final_path"
 	export GOPATH="$final_path/go"
 	export GOCACHE="$final_path/go/.cache"
 	export GOBIN=$GOPATH/bin
-	# Setup nodejs and npm
+	# Setup tensorflow library
+	./scripts/dist/install-tensorflow.sh
 	# Build server from source
 	make all install DESTDIR=$final_path/build 2>&1
 	ynh_secure_remove --file="$final_path/go"