nishchandel commited on
Commit
5b7ca6d
·
1 Parent(s): 93401eb

fixed cache error

Browse files
analysis/audio_analyser.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  import torch
2
  import librosa
3
  import numpy as np
 
1
+ import os
2
+ os.environ['LIBROSA_CACHE_DIR'] = '/tmp'
3
+ os.environ['JOBLIB_TEMP_FOLDER'] = '/tmp'
4
+ os.environ['NUMBA_CACHE_DIR'] = '/tmp'
5
+
6
  import torch
7
  import librosa
8
  import numpy as np
extraction/media_extractor.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
- os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
3
- os.environ['LIBROSA_CACHE_LEVEL'] = '0'
 
4
 
5
  import cv2
6
  import librosa
 
1
  import os
2
+ os.environ['LIBROSA_CACHE_DIR'] = '/tmp'
3
+ os.environ['JOBLIB_TEMP_FOLDER'] = '/tmp'
4
+ os.environ['NUMBA_CACHE_DIR'] = '/tmp'
5
 
6
  import cv2
7
  import librosa
main.py CHANGED
@@ -1,8 +1,8 @@
1
  import os
2
- os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
3
  os.environ['HF_HOME'] = '/tmp/hf_home'
4
- os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
5
- os.environ['LIBROSA_CACHE_LEVEL'] = '0'
 
6
 
7
  from fastapi import FastAPI, File, UploadFile, HTTPException, Query
8
  from fastapi.middleware.cors import CORSMiddleware
 
1
  import os
 
2
  os.environ['HF_HOME'] = '/tmp/hf_home'
3
+ os.environ['LIBROSA_CACHE_DIR'] = '/tmp'
4
+ os.environ['JOBLIB_TEMP_FOLDER'] = '/tmp'
5
+ os.environ['NUMBA_CACHE_DIR'] = '/tmp'
6
 
7
  from fastapi import FastAPI, File, UploadFile, HTTPException, Query
8
  from fastapi.middleware.cors import CORSMiddleware
models/load_models.py CHANGED
@@ -1,5 +1,4 @@
1
  import os
2
- os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
3
  os.environ['HF_HOME'] = '/tmp/hf_home'
4
 
5
  from transformers import AutoModelForImageClassification, AutoImageProcessor
 
1
  import os
 
2
  os.environ['HF_HOME'] = '/tmp/hf_home'
3
 
4
  from transformers import AutoModelForImageClassification, AutoImageProcessor
pipeline.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  from typing import Dict
2
  from extraction.media_extractor import MediaExtractor
3
  from analysis.video_analyser import VideoAnalyzer
 
1
+ import os
2
+ os.environ['LIBROSA_CACHE_DIR'] = '/tmp'
3
+ os.environ['JOBLIB_TEMP_FOLDER'] = '/tmp'
4
+ os.environ['NUMBA_CACHE_DIR'] = '/tmp'
5
+
6
  from typing import Dict
7
  from extraction.media_extractor import MediaExtractor
8
  from analysis.video_analyser import VideoAnalyzer
requirements.txt CHANGED
@@ -3,33 +3,28 @@ fastapi==0.115.5
3
  uvicorn[standard]==0.32.0
4
  python-multipart==0.0.12
5
  pydantic==2.10.2
6
- pydantic-settings==2.6.1
7
 
8
  # Core ML Framework
9
  torch==2.5.1
10
  torchvision==0.20.1
11
- torchaudio==2.5.1
12
  transformers==4.46.3
13
 
14
  # Computer Vision
15
  opencv-python-headless==4.10.0.84
16
  Pillow==11.0.0
17
 
18
- # Audio Processing
19
- librosa==0.10.2.post1
20
  soundfile==0.12.1
21
- ffmpeg-python==0.2.0
22
  audioread==3.0.1
23
 
24
  # LLM Integration
25
  langchain==0.3.7
26
- langchain-core==0.3.15
27
  langchain-google-genai==2.0.5
28
  google-generativeai==0.8.3
29
 
30
  # Data Processing
31
  numpy==1.26.4
32
- pandas==2.2.3
33
  scipy==1.14.1
34
 
35
  # Utilities
 
3
  uvicorn[standard]==0.32.0
4
  python-multipart==0.0.12
5
  pydantic==2.10.2
 
6
 
7
  # Core ML Framework
8
  torch==2.5.1
9
  torchvision==0.20.1
 
10
  transformers==4.46.3
11
 
12
  # Computer Vision
13
  opencv-python-headless==4.10.0.84
14
  Pillow==11.0.0
15
 
16
+ # Audio Processing - Modern compatible versions
17
+ librosa==0.10.1
18
  soundfile==0.12.1
 
19
  audioread==3.0.1
20
 
21
  # LLM Integration
22
  langchain==0.3.7
 
23
  langchain-google-genai==2.0.5
24
  google-generativeai==0.8.3
25
 
26
  # Data Processing
27
  numpy==1.26.4
 
28
  scipy==1.14.1
29
 
30
  # Utilities