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
Files changed (2) hide show
  1. inference_wrapper.py +1 -1
  2. main.py +3 -3
inference_wrapper.py CHANGED
@@ -22,7 +22,7 @@ class StableVITONInference:
22
  """
23
  def __init__(
24
  self,
25
- model_path: str = "fashn-ai/fashn-vton-1.5",
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 local inference engine
39
  model = StableVITONInference(
40
- model_path=os.getenv("MODEL_PATH", "yisol/IDM-VTON"),
41
- device="auto"
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: