IFMedTechdemo commited on
Commit
a8de3ef
·
verified ·
1 Parent(s): 4aadd2a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -6,6 +6,7 @@ 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"
11
  # List of Python files and corresponding class names
 
6
  import torch
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