Update app.py

app.py

@@ -1,8 +1,5 @@
-import gradio as gr
-from PIL import Image
-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
+import torch
+from transformers import AutoTokenizer
 from transformers import (
     CLIPImageProcessor,
     CLIPVisionModelWithProjection,
@@ -10,11 +7,14 @@ from transformers import (
     CLIPTextModelWithProjection,
 )
 from diffusers import DDPMScheduler,AutoencoderKL
+import gradio as gr
+from PIL import Image
+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
 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