Update app.py

app.py

@@ -5,6 +5,7 @@ os.system("pip install huggingface_hub==0.24.7")
 os.system("pip install gradio accelerate==0.25.0 torchmetrics==1.2.1 tqdm==4.66.1 transformers==4.36.2 diffusers==0.25 einops==0.7.0 bitsandbytes==0.39.0 scipy==1.11.1 opencv-python gradio==4.24.0 fvcore cloudpickle omegaconf pycocotools basicsr av onnxruntime==1.16.2 peft==0.11.1 huggingface_hub==0.24.7 --no-deps")
 import gradio as gr
 import torch
+import spaces
 from PIL import Image
 import torch.nn.functional as F
 from transformers import CLIPImageProcessor
@@ -130,7 +131,7 @@ pipe = TryonPipeline.from_pretrained(
             unet_encoder = unet_encoder,
             torch_dtype=torch.float16,
     )
-
+@spaces.GPU
 def generate_virtual_try_on(person_image, cloth_image, mask_image, pose_image,cloth_des):
     pipe.to(device)
     # Prepare the input images as tensors