Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from typing import Tuple, Dict, List
|
|
| 8 |
import cv2
|
| 9 |
from pathlib import Path
|
| 10 |
|
| 11 |
-
from
|
| 12 |
|
| 13 |
class ImageProcessor:
|
| 14 |
def __init__(self, model_path: str = "briaai/RMBG-1.4"):
|
|
|
|
| 8 |
import cv2
|
| 9 |
from pathlib import Path
|
| 10 |
|
| 11 |
+
from briacustom import MultiTargetBriaRMBG, ClothingType, GarmentFeatures
|
| 12 |
|
| 13 |
class ImageProcessor:
|
| 14 |
def __init__(self, model_path: str = "briaai/RMBG-1.4"):
|