Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -8,6 +8,7 @@ from preprocess.openpose.run_openpose import OpenPose
 from preprocess.humanparsing.run_parsing import Parsing
 from ootd.inference_ootd_hd import OOTDiffusionHD
 from ootd.inference_ootd_dc import OOTDiffusionDC
+import spaces
 
 app = Flask(__name__)
 
@@ -26,6 +27,8 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 category_dict = ['upperbody', 'lowerbody', 'dress']
 category_dict_utils = ['upper_body', 'lower_body', 'dresses']
 
+
+@spaces.GPU
 @app.route("/process_hd", methods=["POST"])
 def process_hd():
     data = request.files
@@ -79,6 +82,7 @@ def process_hd():
 
     return jsonify(images=base64_images)
 
+@spaces.GPU
 @app.route("/process_dc", methods=["POST"])
 def process_dc():
     data = request.files