further fix

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 import spaces
 import torch
 from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
@@ -12,6 +13,8 @@ import utils
 from utils import BoundingBox
 import blurnonymize
 
+os.environ['HF_HOME'] = '/data/.huggingface'
+
 MODEL_NAME = "cborg/qwen2.5VL-3b-privacydetector"
 MAX_NEW_TOKENS = 2048
 TEMPERATURE = 1.0

blurnonymize.py

@@ -158,12 +158,11 @@ class ImageBlurnonymizer:
 
         # self.device = "cuda" if torch.cuda.is_available() else "cpu"
         # self.device = "cuda"
-        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        self.device = "cuda"
         # Set the device for PyTorch
-        torch.cuda.set_device(self.device)
         self.predictor = SAM2ImagePredictor.from_pretrained(
             "facebook/sam2.1-hiera-small",
-            device="cuda",
+            device=self.device,
         )
 
     @staticmethod