simple compat for windows + one-click-install/run

README.md

@@ -19,7 +19,7 @@ Thanks to an improved augmentation and training loop, as well as a new dataset,
 
 ## Requirements
 
- - Basic knowledge of a (Linux) terminal
+ - ~~Basic knowledge of a (Linux) terminal~~
  - Working git, python3 and pytorch installation (other dependencies automatically are installed)
 
 ## Inference
@@ -78,4 +78,4 @@ Not only will you get the satisfaction of supporting an individual's passions, b
 
 - [ ] Releasing training code + tutorial
 - [x] HF space! (Thanks to @adolinska!)
-- [ ] Colab demo 
+- [ ] Colab demo 

app.py

@@ -30,6 +30,8 @@ def inference(input_file, **kwargs):
     if USE_CUDA:
         main_model.to('cuda')
         inference_model.to('cuda')
+    else:
+        print('error loading model to cuda')
 
     return "out.wav"
 
@@ -54,4 +56,4 @@ def inference(input_file, **kwargs):
     description = desc,
     article = made,
     theme=gr.themes.Soft(primary_hue="purple",secondary_hue="violet", neutral_hue="neutral")
-).launch()
+).launch(inbrowser=True)

requirements.txt

@@ -1,2 +1,2 @@
 git+https://github.com/peterwilli/audio-maister.git
-numpy==1.23.0
+numpy==1.23.0 

windows-oneclick-webui.bat

@@ -0,0 +1,32 @@
+@echo off
+SET venv_dir=venv
+SET pyfile=main.py
+SET python=%venv_dir%\Scripts\python.exe
+
+REM Check if the virtual environment directory exists
+IF EXIST "%venv_dir%\Scripts\activate.bat" (
+    ECHO Virtual environment found. Activating...
+    CALL %venv_dir%\Scripts\activate.bat
+) ELSE (
+    ECHO Creating virtual environment...
+    python -m venv %venv_dir%
+    CALL %venv_dir%\Scripts\activate.bat
+)
+
+REM Check if the virtual environment is activated
+IF NOT "%VIRTUAL_ENV%" == "" (
+    ECHO Virtual environment activated.
+    ECHO Installing dependencies...
+    %python% -m pip install --upgrade pip
+    %python% -m pip install -r requirements.txt 
+    %python% -m pip install gradio==3.50.2
+    ECHO Dependencies installed.
+
+    REM Run the Python script
+    python app.py
+) ELSE (
+    ECHO Failed to activate virtual environment.
+)
+
+REM Pause the command window
+pause