cache error fixed

extraction/media_extractor.py

@@ -1,15 +1,14 @@
 import os
+os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
+os.environ['LIBROSA_CACHE_LEVEL'] = '0'
+
 import cv2
-from pathlib import Path
 import librosa
 import subprocess
 import numpy as np
 from typing import List, Dict, Tuple
 from extraction.timeline_generator import TimelineGenerator
 
-os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
-os.environ['LIBROSA_CACHE_LEVEL'] = '0'
-
 class MediaExtractor:
     
     def __init__(self, frames_per_interval: int = 5):

main.py

@@ -1,10 +1,15 @@
+import os
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
+os.environ['HF_HOME'] = '/tmp/hf_home'
+os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
+os.environ['LIBROSA_CACHE_LEVEL'] = '0'
+
 from fastapi import FastAPI, File, UploadFile, HTTPException, Query
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from typing import List
 from contextlib import asynccontextmanager
-import os
 import uuid
 import shutil
 import json
@@ -12,11 +17,6 @@ from pathlib import Path
 from datetime import datetime
 from pipeline import DeepfakeDetectionPipeline
 
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
-os.environ['HF_HOME'] = '/tmp/hf_home'
-os.environ['LIBROSA_CACHE_DIR'] = '/tmp/librosa_cache'
-os.environ['LIBROSA_CACHE_LEVEL'] = '0'
-
 analysis_history = []
 MAX_HISTORY = 10
 

models/load_models.py

@@ -1,11 +1,11 @@
-from transformers import AutoModelForImageClassification, AutoImageProcessor
-from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
-import torch
 import os
-
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
 os.environ['HF_HOME'] = '/tmp/hf_home'
 
+from transformers import AutoModelForImageClassification, AutoImageProcessor
+from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
+import torch
+
 class ModelLoader:
     
     _instance = None