Spaces:

theoracle
/

professional_head

Running on Zero

theoracle commited on May 7

Commit

800f45f

verified ·

1 Parent(s): 666f90c

Update background_edit.py

Files changed (1) hide show

background_edit.py CHANGED Viewed

@@ -12,6 +12,7 @@ os.makedirs(yolo_cfg, exist_ok=True)
 os.environ["HF_HOME"]         = hf_home
 os.environ["YOLO_CONFIG_DIR"] = yolo_cfg
 from ultralytics import YOLO
 import numpy as np
 import torch

 os.environ["HF_HOME"]         = hf_home
 os.environ["YOLO_CONFIG_DIR"] = yolo_cfg
+import spaces
 from ultralytics import YOLO
 import numpy as np
 import torch