Update app.py

app.py

@@ -1,3 +1,8 @@
+# Import spaces first to control GPU initialization
+import spaces
+
+# Now import other packages
+import torch
 import gradio as gr
 from PIL import Image
 from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
@@ -11,10 +16,8 @@ from transformers import (
 )
 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
@@ -24,6 +27,8 @@ from preprocess.openpose.run_openpose import OpenPose
 from detectron2.data.detection_utils import convert_PIL_to_numpy, _apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
 
+# Rest of your code remains the same...
+
 
 # Function to convert a PIL image to a binary mask
 def pil_to_binary_mask(pil_image, threshold=0):