Spaces:
Runtime error
Runtime error
Iliya Belous Claude Opus 4.6 commited on
Commit ·
eec31df
1
Parent(s): f7a3af8
fix: switch to omnitry_v1_unified weights (official demo default)
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ logger = logging.getLogger("vton")
|
|
| 32 |
# --- Load OmniTry pipeline at startup ---
|
| 33 |
hf_token = os.environ.get("HF_TOKEN")
|
| 34 |
logger.info("Loading OmniTry pipeline...")
|
| 35 |
-
pipe = load_pipeline(hf_token)
|
| 36 |
logger.info("OmniTry pipeline ready.")
|
| 37 |
|
| 38 |
# Category mapping (backend sends top/bottom/outerwear/dress)
|
|
|
|
| 32 |
# --- Load OmniTry pipeline at startup ---
|
| 33 |
hf_token = os.environ.get("HF_TOKEN")
|
| 34 |
logger.info("Loading OmniTry pipeline...")
|
| 35 |
+
pipe = load_pipeline(hf_token, weight_name="omnitry_v1_unified.safetensors")
|
| 36 |
logger.info("OmniTry pipeline ready.")
|
| 37 |
|
| 38 |
# Category mapping (backend sends top/bottom/outerwear/dress)
|