Spaces:
Runtime error
Runtime error
Upload Multi_Birefnet_for_Background_Removal.bat
Browse files
Multi_Birefnet_for_Background_Removal.bat
ADDED
|
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
@echo off
|
| 2 |
+
pushd %~dp0
|
| 3 |
+
|
| 4 |
+
set BIREFNET_PORT=7860
|
| 5 |
+
set HOME=%USERPROFILE%
|
| 6 |
+
set CUDA_VERSION=
|
| 7 |
+
|
| 8 |
+
:: Check if CUDA 12.1 is installed
|
| 9 |
+
python -c "import torch; print(torch.version.cuda)" 2>nul | findstr /C:"12.1" >nul
|
| 10 |
+
if %errorlevel% == 0 (
|
| 11 |
+
set CUDA_VERSION=12.1
|
| 12 |
+
) else (
|
| 13 |
+
set CUDA_VERSION=none
|
| 14 |
+
)
|
| 15 |
+
|
| 16 |
+
if not exist MBBR_venv\ (
|
| 17 |
+
python -m venv MBBR_venv
|
| 18 |
+
call MBBR_venv\Scripts\activate
|
| 19 |
+
python -m pip install --upgrade pip
|
| 20 |
+
REM pip install Multi_Birefnet_for_Background_Removal
|
| 21 |
+
git clone https://huggingface.co/spaces/NegiTurkey/Multi_Birefnetfor_Background_Removal temp
|
| 22 |
+
REM Move all files and folders from temp to the parent directory
|
| 23 |
+
xcopy temp\* .\
|
| 24 |
+
REM Remove the now-empty temp directory
|
| 25 |
+
rmdir /s /q temp
|
| 26 |
+
if "%CUDA_VERSION%"=="12.1" (
|
| 27 |
+
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121
|
| 28 |
+
powershell -Command "(Get-Content app.py) -replace 'to\(\"cpu\"\)', 'to(\"cuda\")' | Set-Content app.py"
|
| 29 |
+
) else (
|
| 30 |
+
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cpu
|
| 31 |
+
)
|
| 32 |
+
pip install -r requirements.txt
|
| 33 |
+
)
|
| 34 |
+
|
| 35 |
+
call MBBR_venv\Scripts\activate
|
| 36 |
+
start http://localhost:%BIREFNET_PORT%/
|
| 37 |
+
echo http://localhost:%BIREFNET_PORT%/
|
| 38 |
+
|
| 39 |
+
python -m app --port %BIREFNET_PORT%
|
| 40 |
+
REM python -m app --port %BIREFNET_PORT%
|
| 41 |
+
|
| 42 |
+
popd
|
| 43 |
+
|
| 44 |
+
|