nishchandel commited on
Commit
44f01ce
·
1 Parent(s): 1eff516

cache error fixed

Browse files
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