Update app.py

app.py

@@ -1,4 +1,4 @@
-import spaces
+
 import gradio as gr
 from PIL import Image
 from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
@@ -120,10 +120,10 @@ pipe = TryonPipeline.from_pretrained(
         torch_dtype=torch.float16,
 )
 pipe.unet_encoder = UNet_Encoder
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipe.to(device)
+def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
 
-@spaces.GPU
-def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
-    device = "cuda"
     
     openpose_model.preprocessor.body_estimation.model.to(device)
     pipe.to(device)
@@ -177,6 +177,7 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
             with torch.no_grad():
                 prompt = "model is wearing " + garment_des
                 negative_prompt = "monochrome, lowres, bad anatomy, worst quality, low quality"
+            
                 with torch.inference_mode():
                     (
                         prompt_embeds,
@@ -309,5 +310,5 @@ with image_blocks as demo:
             
 
 
-image_blocks.launch()
+demo.launch()