From 191731a4812354f26a7e202758a4e4cb1099976a Mon Sep 17 00:00:00 2001 From: Xavi Vinaixa Date: Mon, 5 Jun 2023 07:37:30 +0200 Subject: [PATCH] Compatibility with Google Colab --- requirements-COLAB.txt | 11 +++++++++++ requirements-GPU.txt | 2 +- requirements.txt | 2 +- 3 files changed, 13 insertions(+), 2 deletions(-) create mode 100644 requirements-COLAB.txt diff --git a/requirements-COLAB.txt b/requirements-COLAB.txt new file mode 100644 index 0000000..b4ec730 --- /dev/null +++ b/requirements-COLAB.txt @@ -0,0 +1,11 @@ +numba==0.57.0 +ffmpeg_python==0.2.0 +gradio==3.33.1 +insightface==0.7.3 +numpy==1.24.3 +onnx==1.14.0 +onnxruntime==1.15.0 +opencv_python==4.7.0.72 +opencv_python_headless==4.7.0.72 +progressbar33==2.4 +scikit-image==0.20.0 \ No newline at end of file diff --git a/requirements-GPU.txt b/requirements-GPU.txt index d0b1204..c1e58c7 100644 --- a/requirements-GPU.txt +++ b/requirements-GPU.txt @@ -7,4 +7,4 @@ onnxruntime_gpu==1.15.0 opencv_python==4.7.0.72 opencv_python_headless==4.7.0.72 progressbar33==2.4 -scikit-image +scikit-image==0.20.0 diff --git a/requirements.txt b/requirements.txt index 1babd63..7473e97 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ onnxruntime==1.15.0 opencv_python==4.7.0.72 opencv_python_headless==4.7.0.72 progressbar33==2.4 -scikit-image +scikit-image==0.20.0