asutermo commited on
Commit
da1d0d5
·
1 Parent(s): 8421d2f
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -20,7 +20,7 @@ class Predictor(BasePredictor):
20
  torch_dtype=self.dtype)
21
 
22
  def predict(self,
23
- hf_token: Secret(description="Hugging Face API token. Create a write token at https://huggingface.co/settings/token. You also need to approve the Flux Dev terms."),
24
  image: Path = Input(description="Image file path", default="https://github.com/nftblackmagic/catvton-flux/raw/main/example/person/1.jpg"),
25
  mask: Path = Input(description="Mask file path", default="https://github.com/nftblackmagic/catvton-flux/blob/main/example/person/1_mask.png?raw=true"),
26
  try_on: bool = Input(True, description="Try on or try off"),
 
20
  torch_dtype=self.dtype)
21
 
22
  def predict(self,
23
+ hf_token: Secret = Input(description="Hugging Face API token. Create a write token at https://huggingface.co/settings/token. You also need to approve the Flux Dev terms."),
24
  image: Path = Input(description="Image file path", default="https://github.com/nftblackmagic/catvton-flux/raw/main/example/person/1.jpg"),
25
  mask: Path = Input(description="Mask file path", default="https://github.com/nftblackmagic/catvton-flux/blob/main/example/person/1_mask.png?raw=true"),
26
  try_on: bool = Input(True, description="Try on or try off"),