Spaces:
Runtime error
Runtime error
Update main.py
Browse files
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 = "
|
| 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 = "
|
| 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 |
|