Update video_enhancer.py

video_enhancer.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 import torch
 import io
+import os
 import asyncio
 from fastapi.responses import StreamingResponse
 from basicsr.archs.rrdbnet_arch import RRDBNet
@@ -9,6 +10,10 @@ from realesrgan import RealESRGANer
 from huggingface_hub import hf_hub_download
 from concurrent.futures import ThreadPoolExecutor
 
+cache_dir = '/tmp/huggingface_cache'
+os.environ['HF_HOME'] = cache_dir
+os.makedirs(cache_dir, exist_ok=True)
+
 class VideoEnhancer:
     def __init__(self, model_name="RealESRGAN_x4plus"):
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')