Update app.py
Browse files
app.py
CHANGED
@@ -122,7 +122,6 @@ def pil_to_binary_mask(pil_image, threshold=0):
|
|
122 |
output_mask = Image.fromarray(mask)
|
123 |
return output_mask
|
124 |
|
125 |
-
@spaces.GPU
|
126 |
def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
127 |
device = "cuda"
|
128 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
@@ -241,6 +240,7 @@ def encode_image_to_base64(image):
|
|
241 |
image.save(buffered, format="PNG")
|
242 |
return base64.b64encode(buffered.getvalue()).decode('utf-8')
|
243 |
|
|
|
244 |
@app.route('/tryon', methods=['POST'])
|
245 |
def tryon():
|
246 |
data = request.json
|
|
|
122 |
output_mask = Image.fromarray(mask)
|
123 |
return output_mask
|
124 |
|
|
|
125 |
def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
|
126 |
device = "cuda"
|
127 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
|
|
240 |
image.save(buffered, format="PNG")
|
241 |
return base64.b64encode(buffered.getvalue()).decode('utf-8')
|
242 |
|
243 |
+
@spaces.GPU
|
244 |
@app.route('/tryon', methods=['POST'])
|
245 |
def tryon():
|
246 |
data = request.json
|