brew install tensorflow for macos

viam-modules · Oct 11, 2024 · 552df1b · 552df1b
1 parent bbc47c8
commit 552df1b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/go/Makefile b/go/Makefile
@@ -64,9 +64,11 @@ else
 endif
 
 setup:
-	if [ "$(UNAME_S)" = "Linux" ]; then \
-		sudo apt install -y vim libjpeg-dev pkg-config patchelf; \
-	fi
+ifeq ($(MOD_OS),Darwin)
+	brew install tensorflowlite;
+else ifeq ($(MOD_OS),Linux)
+	sudo apt install -y vim libjpeg-dev pkg-config patchelf; 
+endif
 
 clean:
 	rm -rf tflite_cpu module.tar.gz