Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files- app.py +4 -1
- requirements.txt +2 -0
app.py
CHANGED
|
@@ -1,9 +1,12 @@
|
|
| 1 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
from nemo.collections.speechlm2.models import SALM
|
| 4 |
import torch
|
| 5 |
import tempfile
|
| 6 |
-
import os
|
| 7 |
|
| 8 |
# Load model using official NVIDIA NeMo approach
|
| 9 |
model_id = "nvidia/canary-qwen-2.5b"
|
|
|
|
| 1 |
|
| 2 |
+
import os
|
| 3 |
+
# Fix OpenMP environment variable issue
|
| 4 |
+
os.environ['OMP_NUM_THREADS'] = '1'
|
| 5 |
+
|
| 6 |
import gradio as gr
|
| 7 |
from nemo.collections.speechlm2.models import SALM
|
| 8 |
import torch
|
| 9 |
import tempfile
|
|
|
|
| 10 |
|
| 11 |
# Load model using official NVIDIA NeMo approach
|
| 12 |
model_id = "nvidia/canary-qwen-2.5b"
|
requirements.txt
CHANGED
|
@@ -2,3 +2,5 @@ torch>=2.6.0
|
|
| 2 |
gradio>=4.0.0
|
| 3 |
nemo_toolkit[asr,tts] @ git+https://github.com/NVIDIA/NeMo.git
|
| 4 |
accelerate>=0.20.0
|
|
|
|
|
|
|
|
|
| 2 |
gradio>=4.0.0
|
| 3 |
nemo_toolkit[asr,tts] @ git+https://github.com/NVIDIA/NeMo.git
|
| 4 |
accelerate>=0.20.0
|
| 5 |
+
sacrebleu>=2.0.0
|
| 6 |
+
omegaconf>=2.3.0
|