From 98dc15f558b14d8504431cabc1667f1adea8e1bc Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Mon, 31 Jul 2023 20:41:48 +0900
Subject: [PATCH] fix: protobuf install

---
 install.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/install.py b/install.py
index 91679b4..fbbf26b 100644
--- a/install.py
+++ b/install.py
@@ -7,7 +7,7 @@ from importlib.metadata import version  # python >= 3.8
 
 from packaging.version import parse
 
-import_name = {"py-cpuinfo": "cpuinfo"}
+import_name = {"py-cpuinfo": "cpuinfo", "protobuf": "google.protobuf"}
 
 
 def is_installed(
@@ -44,11 +44,13 @@ def run_pip(*args):
 def install():
     deps = [
         # requirements
-        ("ultralytics", "8.0.143", None),
+        ("ultralytics", "8.0.145", None),
         ("mediapipe", "0.10.2", None),
         ("rich", "13.4.2", None),
         # ultralytics
         ("py-cpuinfo", None, None),
+        # mediapipe
+        ("protobuf", "3.20", "3.9999"),
     ]
 
     for pkg, low, high in deps: