Commit
·
55285bf
1
Parent(s):
8f209ca
finally done maybe part 2
Browse files- app.py +10 -1
- packages.txt +2 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
import torchaudio
|
| 4 |
-
from transformers import pipeline, AutoModel
|
| 5 |
import librosa
|
| 6 |
import numpy as np
|
| 7 |
import re
|
|
@@ -17,6 +17,15 @@ from functools import lru_cache
|
|
| 17 |
from enum import Enum
|
| 18 |
from huggingface_hub import login, InferenceClient
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
# ============================================
|
| 21 |
# ENVIRONMENT & LOGGING SETUP
|
| 22 |
# ============================================
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
import torchaudio
|
| 4 |
+
from transformers import pipeline, AutoModel, AutoConfig
|
| 5 |
import librosa
|
| 6 |
import numpy as np
|
| 7 |
import re
|
|
|
|
| 17 |
from enum import Enum
|
| 18 |
from huggingface_hub import login, InferenceClient
|
| 19 |
|
| 20 |
+
# Pre-load onnxruntime to handle stack execution issues
|
| 21 |
+
try:
|
| 22 |
+
import onnxruntime as ort
|
| 23 |
+
logger_temp = logging.getLogger("onnxruntime_check")
|
| 24 |
+
logger_temp.info(f"✅ onnxruntime loaded successfully: {ort.__version__}")
|
| 25 |
+
except Exception as e:
|
| 26 |
+
logger_temp = logging.getLogger("onnxruntime_check")
|
| 27 |
+
logger_temp.warning(f"⚠️ onnxruntime import issue: {e}")
|
| 28 |
+
|
| 29 |
# ============================================
|
| 30 |
# ENVIRONMENT & LOGGING SETUP
|
| 31 |
# ============================================
|
packages.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ffmpeg
|
| 2 |
+
libsndfile1
|
requirements.txt
CHANGED
|
@@ -27,7 +27,7 @@ protobuf==3.20.3
|
|
| 27 |
|
| 28 |
# ML Acceleration
|
| 29 |
accelerate==0.25.0
|
| 30 |
-
onnxruntime==1.
|
| 31 |
|
| 32 |
# Video/Media Processing
|
| 33 |
torchcodec==0.1.0
|
|
|
|
| 27 |
|
| 28 |
# ML Acceleration
|
| 29 |
accelerate==0.25.0
|
| 30 |
+
onnxruntime==1.17.1
|
| 31 |
|
| 32 |
# Video/Media Processing
|
| 33 |
torchcodec==0.1.0
|