Spaces:
Sleeping
Sleeping
Commit
·
44f01ce
1
Parent(s):
1eff516
cache error fixed
Browse files- extraction/media_extractor.py +3 -0
- main.py +5 -0
- models/load_models.py +4 -0
extraction/media_extractor.py
CHANGED
|
@@ -7,6 +7,9 @@ import numpy as np
|
|
| 7 |
from typing import List, Dict, Tuple
|
| 8 |
from extraction.timeline_generator import TimelineGenerator
|
| 9 |
|
|
|
|
|
|
|
|
|
|
| 10 |
class MediaExtractor:
|
| 11 |
|
| 12 |
def __init__(self, frames_per_interval: int = 5):
|
|
|
|
| 7 |
from typing import List, Dict, Tuple
|
| 8 |
from extraction.timeline_generator import TimelineGenerator
|
| 9 |
|
| 10 |
+
os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
|
| 11 |
+
os.environ['LIBROSA_CACHE_LEVEL'] = '0'
|
| 12 |
+
|
| 13 |
class MediaExtractor:
|
| 14 |
|
| 15 |
def __init__(self, frames_per_interval: int = 5):
|
main.py
CHANGED
|
@@ -12,6 +12,11 @@ from pathlib import Path
|
|
| 12 |
from datetime import datetime
|
| 13 |
from pipeline import DeepfakeDetectionPipeline
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
analysis_history = []
|
| 16 |
MAX_HISTORY = 10
|
| 17 |
|
|
|
|
| 12 |
from datetime import datetime
|
| 13 |
from pipeline import DeepfakeDetectionPipeline
|
| 14 |
|
| 15 |
+
os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
|
| 16 |
+
os.environ['HF_HOME'] = '/tmp/hf_home'
|
| 17 |
+
os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
|
| 18 |
+
os.environ['LIBROSA_CACHE_LEVEL'] = '0'
|
| 19 |
+
|
| 20 |
analysis_history = []
|
| 21 |
MAX_HISTORY = 10
|
| 22 |
|
models/load_models.py
CHANGED
|
@@ -1,6 +1,10 @@
|
|
| 1 |
from transformers import AutoModelForImageClassification, AutoImageProcessor
|
| 2 |
from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
|
| 3 |
import torch
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
class ModelLoader:
|
| 6 |
|
|
|
|
| 1 |
from transformers import AutoModelForImageClassification, AutoImageProcessor
|
| 2 |
from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
|
| 3 |
import torch
|
| 4 |
+
import os
|
| 5 |
+
|
| 6 |
+
os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
|
| 7 |
+
os.environ['HF_HOME'] = '/tmp/hf_home'
|
| 8 |
|
| 9 |
class ModelLoader:
|
| 10 |
|