nvn04 commited on
Commit
9a4d3b8
·
verified ·
1 Parent(s): 8a94661

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -136,15 +136,15 @@ automasker = AutoMasker(
136
  )
137
 
138
 
139
- # Flux-based CatVTON
140
- access_token = os.getenv("HUGGING_FACE_HUB_TOKEN")
141
- flux_repo = "black-forest-labs/FLUX.1-Fill-dev"
142
- pipeline_flux = FluxTryOnPipeline.from_pretrained(flux_repo, use_auth_token=access_token)
143
- pipeline_flux.load_lora_weights(
144
- os.path.join(repo_path, "flux-lora"),
145
- weight_name='pytorch_lora_weights.safetensors'
146
- )
147
- pipeline_flux.to("cuda", init_weight_dtype(args.mixed_precision))
148
 
149
 
150
 
@@ -369,7 +369,7 @@ def app_gradio():
369
  show_type = gr.Radio(
370
  label="Show Type",
371
  choices=["result only", "input & result", "input & mask & result"],
372
- value="input & mask & result",
373
  )
374
 
375
  # num_inference_steps = 50
 
136
  )
137
 
138
 
139
+ # # Flux-based CatVTON
140
+ # access_token = os.getenv("HUGGING_FACE_HUB_TOKEN")
141
+ # flux_repo = "black-forest-labs/FLUX.1-Fill-dev"
142
+ # pipeline_flux = FluxTryOnPipeline.from_pretrained(flux_repo, use_auth_token=access_token)
143
+ # pipeline_flux.load_lora_weights(
144
+ # os.path.join(repo_path, "flux-lora"),
145
+ # weight_name='pytorch_lora_weights.safetensors'
146
+ # )
147
+ # pipeline_flux.to("cuda", init_weight_dtype(args.mixed_precision))
148
 
149
 
150
 
 
369
  show_type = gr.Radio(
370
  label="Show Type",
371
  choices=["result only", "input & result", "input & mask & result"],
372
+ value="result only",
373
  )
374
 
375
  # num_inference_steps = 50