IFMedTechdemo commited on
Commit
a027eec
·
verified ·
1 Parent(s): d09919c

fix dependencies

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -3,6 +3,8 @@ import numpy as np
3
  import cv2
4
  import os
5
  from huggingface_hub import hf_hub_download
 
 
6
  import importlib.util
7
 
8
  REPO_ID = "IFMedTech/Skin-Analysis"
 
3
  import cv2
4
  import os
5
  from huggingface_hub import hf_hub_download
6
+ import torch
7
+ import segmentation_models_pytorch as smp
8
  import importlib.util
9
 
10
  REPO_ID = "IFMedTech/Skin-Analysis"