Spaces:
Sleeping
Sleeping
Commit
·
5b7ca6d
1
Parent(s):
93401eb
fixed cache error
Browse files- analysis/audio_analyser.py +5 -0
- extraction/media_extractor.py +3 -2
- main.py +3 -3
- models/load_models.py +0 -1
- pipeline.py +5 -0
- requirements.txt +2 -7
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
|
| 3 |
-
os.environ['
|
|
|
|
| 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
|
| 5 |
-
os.environ['
|
|
|
|
| 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.
|
| 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
|