Spaces:
Saad0KH
/
Running on Zero

Saad0KH commited on
Commit
89525cb
·
verified ·
1 Parent(s): bfc2135

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,6 +32,7 @@ models_loaded = False
32
  def load_models():
33
  global unet, tokenizer_one, tokenizer_two, noise_scheduler, text_encoder_one, text_encoder_two
34
  global image_encoder, vae, UNet_Encoder, parsing_model, openpose_model, pipe
 
35
 
36
  if not models_loaded:
37
  base_path = 'yisol/IDM-VTON'
@@ -78,7 +79,6 @@ def load_models():
78
 
79
  models_loaded = True
80
 
81
-
82
  def pil_to_binary_mask(pil_image, threshold=0):
83
  np_image = np.array(pil_image.convert("L")) # Convert to grayscale directly
84
  binary_mask = np_image > threshold
 
32
  def load_models():
33
  global unet, tokenizer_one, tokenizer_two, noise_scheduler, text_encoder_one, text_encoder_two
34
  global image_encoder, vae, UNet_Encoder, parsing_model, openpose_model, pipe
35
+ global models_loaded # Déclarer la variable globale ici
36
 
37
  if not models_loaded:
38
  base_path = 'yisol/IDM-VTON'
 
79
 
80
  models_loaded = True
81
 
 
82
  def pil_to_binary_mask(pil_image, threshold=0):
83
  np_image = np.array(pil_image.convert("L")) # Convert to grayscale directly
84
  binary_mask = np_image > threshold