SmartHeal commited on
Commit
081b143
·
verified ·
1 Parent(s): acd4594

Update src/ai_processor.py

Browse files
Files changed (1) hide show
  1. src/ai_processor.py +3 -0
src/ai_processor.py CHANGED
@@ -16,6 +16,9 @@ import cv2
16
  import numpy as np
17
  from PIL import Image
18
  from PIL.ExifTags import TAGS
 
 
 
19
 
20
  # --- Logging config ---
21
  logging.basicConfig(
 
16
  import numpy as np
17
  from PIL import Image
18
  from PIL.ExifTags import TAGS
19
+ from huggingface_hub import login
20
+ hf_token = os.getenv("hf_token")
21
+ login(token=hf_token)
22
 
23
  # --- Logging config ---
24
  logging.basicConfig(