Update app.py

app.py

@@ -1,7 +1,9 @@
+import os
+import torch
 import gradio as gr
-
-
+import numpy as np
 from PIL import Image
+import spaces  # Move this to the top to avoid CUDA initialization errors
 from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
 from src.unet_hacked_garmnet import UNet2DConditionModel as UNet2DConditionModel_ref
 from src.unet_hacked_tryon import UNet2DConditionModel
@@ -10,24 +12,20 @@ from transformers import (
     CLIPVisionModelWithProjection,
     CLIPTextModel,
     CLIPTextModelWithProjection,
+    AutoTokenizer
 )
-from diffusers import DDPMScheduler,AutoencoderKL
+from diffusers import DDPMScheduler, AutoencoderKL
 from typing import List
-
-import torch
-import os
-from transformers import AutoTokenizer
-import spaces
-import numpy as np
 from utils_mask import get_mask_location
 from torchvision import transforms
 import apply_net
 from preprocess.humanparsing.run_parsing import Parsing
 from preprocess.openpose.run_openpose import OpenPose
-from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
+from detectron2.data.detection_utils import convert_PIL_to_numpy, _apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
 
 
+
 def pil_to_binary_mask(pil_image, threshold=0):
     np_image = np.array(pil_image)
     grayscale_image = Image.fromarray(np_image).convert("L")