Spaces:
Sleeping
Sleeping
StableVITON Deployer commited on
Commit ·
bb682cf
1
Parent(s): f319748
Migration: Switch to Gradio API (merve/fashn-vton-1.5) and updated token
Browse files- inference_wrapper.py +1 -1
- main.py +3 -3
inference_wrapper.py
CHANGED
|
@@ -22,7 +22,7 @@ class StableVITONInference:
|
|
| 22 |
"""
|
| 23 |
def __init__(
|
| 24 |
self,
|
| 25 |
-
model_path: str = "
|
| 26 |
hf_token: Optional[str] = None,
|
| 27 |
**kwargs
|
| 28 |
):
|
|
|
|
| 22 |
"""
|
| 23 |
def __init__(
|
| 24 |
self,
|
| 25 |
+
model_path: str = "merve/fashn-vton-1.5",
|
| 26 |
hf_token: Optional[str] = None,
|
| 27 |
**kwargs
|
| 28 |
):
|
main.py
CHANGED
|
@@ -35,10 +35,10 @@ async def lifespan(app: FastAPI):
|
|
| 35 |
global model
|
| 36 |
try:
|
| 37 |
logger.info("Initializing StableVITON Inference Engine (Lifespan Startup)...")
|
| 38 |
-
# Initialize the
|
| 39 |
model = StableVITONInference(
|
| 40 |
-
model_path=os.getenv("MODEL_PATH", "
|
| 41 |
-
|
| 42 |
)
|
| 43 |
logger.info("StableVITON Inference Engine initialized successfully")
|
| 44 |
except Exception:
|
|
|
|
| 35 |
global model
|
| 36 |
try:
|
| 37 |
logger.info("Initializing StableVITON Inference Engine (Lifespan Startup)...")
|
| 38 |
+
# Initialize the API inference engine
|
| 39 |
model = StableVITONInference(
|
| 40 |
+
model_path=os.getenv("MODEL_PATH", "merve/fashn-vton-1.5"),
|
| 41 |
+
hf_token=os.getenv("HF_TOKEN")
|
| 42 |
)
|
| 43 |
logger.info("StableVITON Inference Engine initialized successfully")
|
| 44 |
except Exception:
|