Update src/ai_processor.py

src/ai_processor.py

@@ -16,9 +16,21 @@ import cv2
 import numpy as np
 from PIL import Image
 from PIL.ExifTags import TAGS
-from huggingface_hub import login
-hf_token = os.getenv("hf_token")
-login(token=hf_token)
+import os, logging
+from huggingface_hub import HfFolder
+
+# Read from env (prefer standard uppercase)
+HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("hf_token")
+
+if HF_TOKEN:
+    # Persist for the ubuntu user so HF/Transformers can reuse it
+    HfFolder.save_token(HF_TOKEN)
+    # Also keep it in-process for libraries that accept a token kwarg
+    os.environ["HF_TOKEN"] = HF_TOKEN
+    logging.info("✅ Hugging Face token configured without interactive login.")
+else:
+    logging.warning("⚠️ HF_TOKEN not set. Set it in /etc/default/smartheal for private/gated models.")
+
 
 # --- Logging config ---
 logging.basicConfig(