Adilmar commited on
Commit
636dfd7
·
verified ·
1 Parent(s): 29a986e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,7 +8,7 @@ import segmentation_models_pytorch as smp
8
  import importlib.util
9
  #import onnxruntime as ort
10
 
11
- REPO_ID = "IFMedTech/Skin-Analysis"
12
  # List of Python files and corresponding class names
13
  PY_MODULES = {
14
  "dark_circles.py": "DarkCircleDetector",
@@ -37,7 +37,7 @@ PY_MODULES = {
37
  # device = "cuda" if torch.cuda.is_available() else "cpu"
38
  # return session, device
39
  def load_model(token):
40
- repo_id = "IFMedTech/Skin-Analysis"
41
  filename = "model/wrinkles_unet_v1.pth"
42
  # token = os.environ.get("HUGGINGFACE_HUB_TOKEN") # Set this env var with your token
43
 
 
8
  import importlib.util
9
  #import onnxruntime as ort
10
 
11
+ REPO_ID = "Adilmar/Skin-Analysis"
12
  # List of Python files and corresponding class names
13
  PY_MODULES = {
14
  "dark_circles.py": "DarkCircleDetector",
 
37
  # device = "cuda" if torch.cuda.is_available() else "cpu"
38
  # return session, device
39
  def load_model(token):
40
+ repo_id = "Adilmar/Skin-Analysis"
41
  filename = "model/wrinkles_unet_v1.pth"
42
  # token = os.environ.get("HUGGINGFACE_HUB_TOKEN") # Set this env var with your token
43