prismaudio-project commited on
Commit
6cba118
·
1 Parent(s): 95a3397
Files changed (2) hide show
  1. app.py +6 -1
  2. setup.sh +0 -1
app.py CHANGED
@@ -2,8 +2,13 @@ import os
2
  # ⭐ Must be set before importing gradio
3
  import subprocess
4
  import sys
 
 
 
 
5
  import spaces
6
- subprocess.run(["bash", "setup.sh"], check=True)
 
7
  os.environ["JAX_PLATFORMS"] = "cpu"
8
  import gradio as gr
9
  import logging
 
2
  # ⭐ Must be set before importing gradio
3
  import subprocess
4
  import sys
5
+ if os.environ.get("SETUP_DONE") != "1":
6
+ subprocess.run(["bash", "setup.sh"], check=True)
7
+ os.environ["SETUP_DONE"] = "1"
8
+ os.execv(sys.executable, [sys.executable] + sys.argv)
9
  import spaces
10
+
11
+
12
  os.environ["JAX_PLATFORMS"] = "cpu"
13
  import gradio as gr
14
  import logging
setup.sh CHANGED
@@ -5,4 +5,3 @@ cd ..
5
  pip install -r scripts/PrismAudio/setup/requirements.txt
6
  pip install tensorflow-cpu==2.15.0
7
  pip install facenet_pytorch==2.6.0 --no-deps
8
- pip install --force-reinstall torch==2.6.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
 
5
  pip install -r scripts/PrismAudio/setup/requirements.txt
6
  pip install tensorflow-cpu==2.15.0
7
  pip install facenet_pytorch==2.6.0 --no-deps