JustNikunj commited on
Commit
55285bf
·
1 Parent(s): 8f209ca

finally done maybe part 2

Browse files
Files changed (3) hide show
  1. app.py +10 -1
  2. packages.txt +2 -0
  3. 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.16.3
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