Spaces:
Running on Zero
Running on Zero
[Admin maintenance] Migrate to ZeroGPU
Browse files
app.py
CHANGED
|
@@ -7,10 +7,13 @@ _omp = os.environ.get("OMP_NUM_THREADS", "")
|
|
| 7 |
if _omp and not _omp.isdigit():
|
| 8 |
os.environ["OMP_NUM_THREADS"] = str(max(1, int("".join(c for c in _omp if c.isdigit()) or "1")))
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
from nemo.collections.asr.models import ASRModel
|
| 11 |
import torch
|
| 12 |
import gradio as gr
|
| 13 |
-
import spaces
|
| 14 |
import gc
|
| 15 |
import shutil
|
| 16 |
from pathlib import Path
|
|
|
|
| 7 |
if _omp and not _omp.isdigit():
|
| 8 |
os.environ["OMP_NUM_THREADS"] = str(max(1, int("".join(c for c in _omp if c.isdigit()) or "1")))
|
| 9 |
|
| 10 |
+
# initializing numba with CUDA breaks ZeroGPU
|
| 11 |
+
os.environ.setdefault("NUMBA_DISABLE_CUDA", "1")
|
| 12 |
+
|
| 13 |
+
import spaces
|
| 14 |
from nemo.collections.asr.models import ASRModel
|
| 15 |
import torch
|
| 16 |
import gradio as gr
|
|
|
|
| 17 |
import gc
|
| 18 |
import shutil
|
| 19 |
from pathlib import Path
|