Update app.py

app.py

@@ -10,7 +10,6 @@ import io
 import logging
 import gradio as gr
 import numpy as np
-import spaces
 import uuid
 import random
 from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
@@ -31,7 +30,15 @@ from preprocess.humanparsing.run_parsing import Parsing
 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
-
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+            return wrapper
 app = Flask(__name__)
 
 # Chemins de base pour les modèles