Saad0KH commited on
Commit
e139ea7
1 Parent(s): 7de42a9

Update run/gradio_ootd.py

Browse files
Files changed (1) hide show
  1. run/gradio_ootd.py +1 -0
run/gradio_ootd.py CHANGED
@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify , send_file
2
  import torch
3
  from PIL import Image, ImageOps
4
  import base64
 
5
  from utils_ootd import get_mask_location
6
  from preprocess.openpose.run_openpose import OpenPose
7
  from preprocess.humanparsing.run_parsing import Parsing
 
2
  import torch
3
  from PIL import Image, ImageOps
4
  import base64
5
+ from io import BytesIO
6
  from utils_ootd import get_mask_location
7
  from preprocess.openpose.run_openpose import OpenPose
8
  from preprocess.humanparsing.run_parsing import Parsing