Spaces:
Running on Zero
Running on Zero
prismaudio-project commited on
Commit ·
6cba118
1
Parent(s): 95a3397
fix
Browse files
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 |
-
|
|
|
|
| 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
|
|
|