Update app.py
Browse files
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
|