Update app.py

app.py

@@ -1,18 +1,23 @@
 import argparse
 import os
+os.environ['CUDA_HOME'] = '/usr/local/cuda'
+os.environ['PATH'] = os.environ['PATH'] + ':/usr/local/cuda/bin'
 from datetime import datetime
 
 import gradio as gr
+import spaces
 import numpy as np
 import torch
 from diffusers.image_processor import VaeImageProcessor
 from huggingface_hub import snapshot_download
 from PIL import Image
-
+torch.jit.script = lambda f: f
 from model.cloth_masker import AutoMasker, vis_mask
-from model.pipeline import CatVTONPipeline
+from model.pipeline import CatVTONPipeline, CatVTONPix2PixPipeline
+from model.flux.pipeline_flux_tryon import FluxTryOnPipeline
 from utils import init_weight_dtype, resize_and_crop, resize_and_padding
 
+
 def parse_args():
     parser = argparse.ArgumentParser(description="Simple example of a training script.")
     parser.add_argument(