From c83dd8c53a504e68510f5fdda26e858024cb61d8 Mon Sep 17 00:00:00 2001
From: predprey2 <44527563+predprey2@users.noreply.github.com>
Date: Tue, 23 May 2023 05:46:23 +0800
Subject: [PATCH] Update install.py

Fixes library names so launch.is_installed() properly checks for installed packages
---
 install.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/install.py b/install.py
index 2f7bd93..ecedd27 100644
--- a/install.py
+++ b/install.py
@@ -2,12 +2,12 @@ import launch
 
 packages = {
     "onnx": "onnx",
-    "onnxruntime-gpu": "onnxruntime-gpu==1.14.0",
-    "opencv-python": "opencv-python",
+    "onnxruntime": "onnxruntime-gpu==1.14.0",
+    "cv2": "opencv-python",
     "numpy": "numpy",
-    "Pillow": "Pillow",
-    "segmentation-refinement": "segmentation-refinement",
-    "scikit-learn": "scikit-learn",
+    "PIL": "Pillow",
+    "segmentation_refinement": "segmentation-refinement",
+    "sklearn": "scikit-learn",
     "clip": "clip",
 }