Updates - app.py

app.py

@@ -7,23 +7,10 @@ Automatically generated by Colaboratory.
 # !rm -rf sample_data
 # !rm -rf fashion-eye-try-on/
 
-import sys
-from threading import Thread
-import gradio as gr
-import torch
-from collections import OrderedDict
-from PIL import Image
-import torch.nn.functional as F
-import torchvision.transforms as transforms
-from cloth_segmentation.networks import U2NET
-import gdown
-from os.path import exists, join, basename, splitext
-import subprocess
-import os
 BASE_DIR = "/home/user/app/fashion-eye-try-on"
 
-os.system(
-    f"git clone https://huggingface.co/spaces/sidharthism/fashion-eye-try-on {BASE_DIR}")
+import os
+os.system(f"git clone https://huggingface.co/spaces/sidharthism/fashion-eye-try-on {BASE_DIR}")
 
 # !pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
 # !pip install -r /content/fashion-eye-try-on/requirements.txt
@@ -34,31 +21,22 @@ os.system(f"cd {BASE_DIR}")
 
 # Download and save checkpoints for cloth mask generation
 os.system(f"rm -rf {BASE_DIR}/cloth_segmentation/checkpoints/")
-os.system(
-    f"gdown --id 1mhF3yqd7R-Uje092eypktNl-RoZNuiCJ -O {BASE_DIR}/cloth_segmentation/checkpoints/")
-
-os.system(
-    f"git clone https://github.com/shadow2496/VITON-HD {BASE_DIR}/VITON-HD")
-
-# checkpoints
-os.system(
-    f"gdown 1RM4OthSM6V4r7kWCu8SbPIPY14Oz8B2u -O {BASE_DIR}/VITON-HD/checkpoints/alias_final.pth")
-os.system(
-    f"gdown 1MBHBddaAs7sy8W40jzLmNL83AUh035F1 -O {BASE_DIR}/VITON-HD/checkpoints/gmm_final.pth")
-os.system(
-    f"gdown 1MBHBddaAs7sy8W40jzLmNL83AUh035F1 -O {BASE_DIR}/VITON-HD/checkpoints/gmm_final.pth")
-os.system(
-    f"gdown 17U1sooR3mVIbe8a7rZuFIF3kukPchHfZ -O {BASE_DIR}/VITON-HD/checkpoints/seg_final.pth")
-# test data
-os.system(
-    f"gdown 1ncEHn_6liOot8sgt3A2DOFJBffvx8tW8 -O {BASE_DIR}/VITON-HD/datasets/test_pairs.txt")
-os.system(
-    f"gdown 1ZA2C8yMOprwc0TV4hvrt0X-ljZugrClq -O {BASE_DIR}/VITON-HD/datasets/test.zip")
-
-os.system(
-    f"unzip {BASE_DIR}/VITON-HD/datasets/test.zip -d {BASE_DIR}/VITON-HD/datasets/")
-
-# @title To clear all the already existing test data
+os.system(f"gdown --id 1mhF3yqd7R-Uje092eypktNl-RoZNuiCJ -O {BASE_DIR}/cloth_segmentation/checkpoints/")
+
+os.system(f"git clone https://github.com/shadow2496/VITON-HD {BASE_DIR}/VITON-HD")
+
+#checkpoints
+os.system(f"gdown 1RM4OthSM6V4r7kWCu8SbPIPY14Oz8B2u -O {BASE_DIR}/VITON-HD/checkpoints/alias_final.pth")
+os.system(f"gdown 1MBHBddaAs7sy8W40jzLmNL83AUh035F1 -O {BASE_DIR}/VITON-HD/checkpoints/gmm_final.pth")
+os.system(f"gdown 1MBHBddaAs7sy8W40jzLmNL83AUh035F1 -O {BASE_DIR}/VITON-HD/checkpoints/gmm_final.pth")
+os.system(f"gdown 17U1sooR3mVIbe8a7rZuFIF3kukPchHfZ -O {BASE_DIR}/VITON-HD/checkpoints/seg_final.pth")
+#test data
+os.system(f"gdown 1ncEHn_6liOot8sgt3A2DOFJBffvx8tW8 -O {BASE_DIR}/VITON-HD/datasets/test_pairs.txt")
+os.system(f"gdown 1ZA2C8yMOprwc0TV4hvrt0X-ljZugrClq -O {BASE_DIR}/VITON-HD/datasets/test.zip")
+
+os.system(f"unzip {BASE_DIR}/VITON-HD/datasets/test.zip -d {BASE_DIR}/VITON-HD/datasets/")
+
+#@title To clear all the already existing test data
 # !rm -rf /content/fashion-eye-try-on/VITON-HD/datasets/test/image
 # !rm -rf /content/fashion-eye-try-on/VITON-HD/datasets/test/image-parse
 # !rm -rf /content/fashion-eye-try-on/VITON-HD/datasets/test/cloth
@@ -66,10 +44,13 @@ os.system(
 # !rm -rf /content/fashion-eye-try-on/VITON-HD/datasets/test/openpose-img
 # !rm -rf /content/fashion-eye-try-on/VITON-HD/datasets/test/openpose-json
 
-"""Paddle"""
+"""Paddle
 
-os.system(
-    f"git clone https://huggingface.co/spaces/sidharthism/pipeline_paddle {BASE_DIR}/pipeline_paddle")
+
+
+"""
+
+os.system(f"git clone https://huggingface.co/spaces/sidharthism/pipeline_paddle {BASE_DIR}/pipeline_paddle")
 
 # Required for paddle and gradio (Jinja2 dependency)
 os.system("pip install paddlepaddle-gpu pymatting")
@@ -77,11 +58,10 @@ os.system(f"pip install -r {BASE_DIR}/pipeline_paddle/requirements.txt")
 
 os.system(f"rm -rf {BASE_DIR}/pipeline_paddle/models")
 if not os.path.exists(f"{BASE_DIR}/pipeline_paddle/models/ppmatting-hrnet_w18-human_1024.pdparams"):
-    if not os.path.exists(f"{BASE_DIR}/pipeline_paddle/models"):
-        os.mkdir(f"{BASE_DIR}/pipeline_paddle/models")
-    os.system(
-        f"wget https://paddleseg.bj.bcebos.com/matting/models/ppmatting-hrnet_w18-human_1024.pdparams -O {BASE_DIR}/pipeline_paddle/models/ppmatting-hrnet_w18-human_1024.pdparams")
-    # !wget "https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz" -O "/content/fashion-eye-try-on/pipeline_paddle/models/hrnet_w18_ssld.tar.gz"
+  if not os.path.exists(f"{BASE_DIR}/pipeline_paddle/models"):
+      os.mkdir(f"{BASE_DIR}/pipeline_paddle/models")
+  os.system(f"wget https://paddleseg.bj.bcebos.com/matting/models/ppmatting-hrnet_w18-human_1024.pdparams -O {BASE_DIR}/pipeline_paddle/models/ppmatting-hrnet_w18-human_1024.pdparams")
+  # !wget "https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz" -O "/content/fashion-eye-try-on/pipeline_paddle/models/hrnet_w18_ssld.tar.gz"
 
 """Initialization
 
@@ -91,44 +71,44 @@ Pose estimator - open pose
 # Clone openpose model repo
 # os.system(f"git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git {BASE_DIR}/openpose")
 
+!rm -rf /content/fashion-eye-try-on/openpose
+!rm /content/cmake-3.13.0-Linux-x86_64.tar.gz
 
-# @ Building and Installation of openpose model
+#@ Building and Installation of openpose model
+import os
+import subprocess
+from os.path import exists, join, basename, splitext
 
 
 project_name = f"{BASE_DIR}/openpose"
 print(project_name)
 if not exists(project_name):
-    # see: https://github.com/CMU-Perceptual-Computing-Lab/openpose/issues/949
-    # install new CMake becaue of CUDA10
-    os.system(
-        f"wget -q https://cmake.org/files/v3.13/cmake-3.13.0-Linux-x86_64.tar.gz")
-    os.system(
-        f"tar xfz cmake-3.13.0-Linux-x86_64.tar.gz --strip-components=1 -C /usr/local")
-    # clone openpose
-    os.system(
-        f"cd {BASE_DIR} && git clone -q --depth 1 https://github.com/CMU-Perceptual-Computing-Lab/openpose.git")
-    os.system(
-        "sed -i 's/execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/execute_process(COMMAND git checkout f019d0dfe86f49d1140961f8c7dec22130c83154 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/g' %s/openpose/CMakeLists.txt" % (BASE_DIR, ))
-    # install system dependencies
-    os.system("apt-get -qq install -y libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev opencl-headers ocl-icd-opencl-dev libviennacl-dev")
-    # build openpose
-    print("Building openpose ... May take nearly 15 mins to build ...")
-    os.system(f"cd {BASE_DIR}/openpose && rm -rf {BASE_DIR}/openpose/build || true && mkdir {BASE_DIR}/openpose/build && cd {BASE_DIR}/openpose/build && cmake .. && make -j`nproc`")
-    print("Openpose successfully build and installed.")
-    # subprocess.Popen(f"cd {BASE_DIR}/openpose && rm -rf {BASE_DIR}/openpose/build || true && mkdir {BASE_DIR}/openpose/build && cd {BASE_DIR}/openpose/build && cmake .. && make -j`nproc`")
-    # subprocess.call(["cd", f"{BASE_DIR}/openpose"])
-    # subprocess.check_output(["rm", "-rf", f"{BASE_DIR}/openpose/build || true"])
-    # subprocess.check_output(["mkdir", f"{BASE_DIR}/openpose/build"])
-    # subprocess.check_output(["cd", f"{BASE_DIR}/openpose/build"])
-    # subprocess.check_output(["cmake", ".."])
-    # subprocess.check_output(["make","-j`nproc`"])
+  # see: https://github.com/CMU-Perceptual-Computing-Lab/openpose/issues/949
+  # install new CMake becaue of CUDA10
+  os.system(f"wget -q https://cmake.org/files/v3.13/cmake-3.13.0-Linux-x86_64.tar.gz")
+  os.system(f"tar xfz cmake-3.13.0-Linux-x86_64.tar.gz --strip-components=1 -C /usr/local")
+  # clone openpose
+  os.system(f"cd {BASE_DIR} && git clone -q --depth 1 https://github.com/CMU-Perceptual-Computing-Lab/openpose.git")
+  os.system("sed -i 's/execute_process(COMMAND git checkout master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/execute_process(COMMAND git checkout f019d0dfe86f49d1140961f8c7dec22130c83154 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/g' %s/openpose/CMakeLists.txt" % (BASE_DIR, ))
+  # install system dependencies
+  os.system("apt-get -qq install -y libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev opencl-headers ocl-icd-opencl-dev libviennacl-dev")
+  # build openpose
+  print("Building openpose ... May take nearly 15 mins to build ...")
+  os.system(f"cd {BASE_DIR}/openpose && rm -rf {BASE_DIR}/openpose/build || true && mkdir {BASE_DIR}/openpose/build && cd {BASE_DIR}/openpose/build && cmake .. && make -j`nproc`")
+  print("Openpose successfully build and installed.")
+  # subprocess.Popen(f"cd {BASE_DIR}/openpose && rm -rf {BASE_DIR}/openpose/build || true && mkdir {BASE_DIR}/openpose/build && cd {BASE_DIR}/openpose/build && cmake .. && make -j`nproc`")
+  # subprocess.call(["cd", f"{BASE_DIR}/openpose"])
+  # subprocess.check_output(["rm", "-rf", f"{BASE_DIR}/openpose/build || true"])
+  # subprocess.check_output(["mkdir", f"{BASE_DIR}/openpose/build"])
+  # subprocess.check_output(["cd", f"{BASE_DIR}/openpose/build"])
+  # subprocess.check_output(["cmake", ".."])
+  # subprocess.check_output(["make","-j`nproc`"])
 
 # !cd {BASE_DIR}/openpose && rm -rf {BASE_DIR}/openpose/build || true && mkdir {BASE_DIR}/openpose/build && cd {BASE_DIR}/openpose/build && cmake .. && make -j`nproc`
 
 """Self correction human parsing"""
 
-os.system(
-    f"git clone https://github.com/PeikeLi/Self-Correction-Human-Parsing.git {BASE_DIR}/human_parse")
+os.system(f"git clone https://github.com/PeikeLi/Self-Correction-Human-Parsing.git {BASE_DIR}/human_parse")
 
 os.system(f"cd {BASE_DIR}/human_parse")
 os.system(f"mkdir {BASE_DIR}/human_parse/checkpoints")
@@ -137,6 +117,7 @@ os.system(f"mkdir {BASE_DIR}/human_parse/checkpoints")
 
 dataset = 'lip'
 
+import gdown
 
 dataset_url = 'https://drive.google.com/uc?id=1k4dllHpu0bdx38J7H28rVVLpU-kOHmnH'
 output = f'{BASE_DIR}/human_parse/checkpoints/final.pth'
@@ -145,19 +126,19 @@ gdown.download(dataset_url, output, quiet=False)
 # For human parse
 os.system("pip install ninja")
 
-"""Preprocessing"""
+"""Preprocessing
 
-# png to jpg
 
+"""
 
+# png to jpg
 def convert_to_jpg(path):
     from PIL import Image
     import os
     if os.path.exists(path):
-        cl = Image.open(path)
-        jpg_path = path[:-4] + ".jpg"
-        cl.save(jpg_path)
-
+      cl = Image.open(path)
+      jpg_path = path[:-4] + ".jpg"
+      cl.save(jpg_path)
 
 def resize_img(path):
     from PIL import Image
@@ -166,19 +147,13 @@ def resize_img(path):
     im = im.resize((768, 1024), Image.BICUBIC)
     im.save(path)
 
-
 def remove_ipynb_checkpoints():
     import os
-    os.system(
-        f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/image/.ipynb_checkpoints")
-    os.system(
-        f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/cloth/.ipynb_checkpoints")
-    os.system(
-        f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/cloth-mask/.ipynb_checkpoints")
+    os.system(f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/image/.ipynb_checkpoints")
+    os.system(f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/cloth/.ipynb_checkpoints")
+    os.system(f"rm -rf {BASE_DIR}/VITON-HD/datasets/test/cloth-mask/.ipynb_checkpoints")
 
 # os.chdir('/content/fashion-eye-try-on')
-
-
 def preprocess():
     remove_ipynb_checkpoints()
     for path in os.listdir(f'{BASE_DIR}/VITON-HD/datasets/test/image/'):
@@ -188,13 +163,12 @@ def preprocess():
     # for path in os.listdir('/content/fashion-eye-try-on/VITON-HD/datasets/test/cloth-mask/'):
     #   resize_img(f'/content/fashion-eye-try-on/VITON-HD/datasets/test/cloth-mask/{path}')
 
+"""Paddle - removing background
 
-"""Paddle - removing background"""
+"""
 
 # PPMatting hrnet 1024
 # --fg_estimate True - for higher quality output but slower prediction
-
-
 def upload_remove_background_and_save_person_image(person_img):
     # !export CUDA_VISIBLE_DEVICES=0
     person_img = person_img.resize((768, 1024), Image.BICUBIC)
@@ -210,7 +184,7 @@ def upload_remove_background_and_save_person_image(person_img):
         --background 'w' \
         --save_dir {BASE_DIR}/VITON-HD/datasets/test/image \
         --fg_estimate True")
-    # --save_dir /content/fashion-eye-try-on/pipeline_paddle/output \
+        # --save_dir /content/fashion-eye-try-on/pipeline_paddle/output \
     try:
         convert_to_jpg(f"{BASE_DIR}/VITON-HD/datasets/test/image/person.png")
         # os.remove("/content/fashion-eye-try-on/pipeline_paddle/output/person_alpha.png")
@@ -222,14 +196,11 @@ def upload_remove_background_and_save_person_image(person_img):
         print(e)
         os.system(f"cd {BASE_DIR}")
 
-
-# @title If multiple GPU available,uncomment and try this code
+#@title If multiple GPU available,uncomment and try this code
 os.system("export CUDA_VISIBLE_DEVICES=0")
 
 # Openpose pose estimation
 # Ubuntu and Mac
-
-
 def estimate_pose():
     os.system(f"cd {BASE_DIR}/openpose && ./build/examples/openpose/openpose.bin --image_dir {BASE_DIR}/VITON-HD/datasets/test/image --write_json {BASE_DIR}/VITON-HD/datasets/test/openpose-json/ --display 0 --face --hand --render_pose 0")
     os.system(f"cd {BASE_DIR}/openpose && ./build/examples/openpose/openpose.bin --image_dir {BASE_DIR}/VITON-HD/datasets/test/image --write_images {BASE_DIR}/VITON-HD/datasets/test/openpose-img/ --display 0 --hand --render_pose 1 --disable_blending true")
@@ -238,8 +209,6 @@ def estimate_pose():
 
 # Run self correction human parser
 # !python3 /content/fashion-eye-try-on/human_parse/simple_extractor.py --dataset 'lip' --model-restore '/content/fashion-eye-try-on/human_parse/checkpoints/final.pth' --input-dir '/content/fashion-eye-try-on/image' --output-dir '/content/fashion-eye-try-on/VITON-HD/datasets/test/image-parse'
-
-
 def generate_human_segmentation_map():
     # remove_ipynb_checkpoints()
     os.system(f"python3 {BASE_DIR}/human_parse/simple_extractor.py --dataset 'lip' --model-restore '{BASE_DIR}/human_parse/checkpoints/final.pth' --input-dir '{BASE_DIR}/VITON-HD/datasets/test/image' --output-dir '{BASE_DIR}/VITON-HD/datasets/test/image-parse'")
@@ -251,25 +220,29 @@ def generate_human_segmentation_map():
 # with open('/content/fashion-eye-try-on/VITON-HD/datasets/test_pairs.txt', 'w') as file:
 #     for model, cloth in pairs:
 #         file.write(f"{model} {cloth}\n")
-
-
 def generate_test_pairs_txt():
     with open(f"{BASE_DIR}/VITON-HD/datasets/test_pairs.txt", 'w') as file:
         file.write(f"person.jpg cloth.jpg\n")
 
 # VITON-HD
 # Transfer the cloth to the model
-
-
 def generate_viton_hd():
     os.system(f"python {BASE_DIR}/VITON-HD/test.py --name output --dataset_list {BASE_DIR}/VITON-HD/datasets/test_pairs.txt  --dataset_dir {BASE_DIR}/VITON-HD/datasets/ --checkpoint_dir {BASE_DIR}/VITON-HD/checkpoints --save_dir {BASE_DIR}/")
 
-
+import sys
 # To resolve ModuleNotFoundError during imports
 if BASE_DIR not in sys.path:
     sys.path.append(BASE_DIR)
     sys.path.append(f"{BASE_DIR}/cloth_segmentation")
 
+from cloth_segmentation.networks import U2NET
+import torchvision.transforms as transforms
+import torch.nn.functional as F
+import os
+from PIL import Image
+from collections import OrderedDict
+
+import torch
 
 device = 'cuda' if torch.cuda.is_available() else "cpu"
 
@@ -405,78 +378,73 @@ def generate_cloth_mask(img_dir, output_dir, chkpt_dir):
             output_img = output_img.convert('L')
             output_img.save(os.path.join(result_dir, image_name[:-4]+'.jpg'))
 
-
 os.system(f"cd {BASE_DIR}")
-
-
+from PIL import Image
 def upload_resize_generate_cloth_mask_and_move_to_viton_hd_test_inputs(cloth_img):
     os.system(f"cd {BASE_DIR}")
     cloth_img = cloth_img.resize((768, 1024), Image.BICUBIC)
     cloth_img.save(f"{BASE_DIR}/cloth/cloth.jpg")
     cloth_img.save(f"{BASE_DIR}/VITON-HD/datasets/test/cloth/cloth.jpg")
     try:
-        generate_cloth_mask(f"{BASE_DIR}/cloth", f"{BASE_DIR}/cloth_mask",
-                            f"{BASE_DIR}/cloth_segmentation/checkpoints/cloth_segm_u2net_latest.pth")
+        generate_cloth_mask(f"{BASE_DIR}/cloth", f"{BASE_DIR}/cloth_mask", f"{BASE_DIR}/cloth_segmentation/checkpoints/cloth_segm_u2net_latest.pth")
         cloth_mask_img = Image.open(f"{BASE_DIR}/cloth_mask/cloth.jpg")
-        cloth_mask_img.save(
-            f"{BASE_DIR}/VITON-HD/datasets/test/cloth-mask/cloth.jpg")
+        cloth_mask_img.save(f"{BASE_DIR}/VITON-HD/datasets/test/cloth-mask/cloth.jpg")
     except Exception as e:
-        print(e)
-
+      print(e)
 
 # Gradio
 os.system("pip install gradio")
 
+import gradio as gr
 # import cv2
-IMAGEPATH = '/content/fashion-eye-try-on/VITON-HD/datasets/test/image'
-CLOTHPATH = '/content/fashion-eye-try-on/VITON-HD/datasets/test/cloth'
-CLOTHMASKPATH = '/content/fashion-eye-try-on/VITON-HD/datasets/test/image'
+from PIL import Image
+IMAGEPATH='/content/fashion-eye-try-on/VITON-HD/datasets/test/image'
+CLOTHPATH='/content/fashion-eye-try-on/VITON-HD/datasets/test/cloth'
+CLOTHMASKPATH='/content/fashion-eye-try-on/VITON-HD/datasets/test/image'
 
+from threading import Thread
 
 def fashion_eye_tryon(person_img, cloth_img):
-    result_img = person_img
-    # img.save(IMAGEPATH + "person.jpg")
-    # dress.save(CLOTHPATH + "cloth.jpg")
-
-    # txt = open("/content/VITON-HD/datasets/test_pairs.txt", "a")
-    # txt.write("person_img.jpg dress_img.jpg\n")
-    # txt.close()
-    # # result
-    # print(person_img.info, cloth_img.info)
-    # p_t1 = Thread(target=upload_remove_background_and_save_person_image, args=(person_img, ))
-    # c_t2 = Thread(target=upload_resize_generate_cloth_mask_and_move_to_viton_hd_test_inputs, args=(cloth_img, ))
-    # p_t1.start()
-    # c_t2.start()
-    # p_t1.join()
-    # c_t2.join()
-    # Estimate pose
-    try:
-        upload_resize_generate_cloth_mask_and_move_to_viton_hd_test_inputs(
-            cloth_img)
-        upload_remove_background_and_save_person_image(person_img)
-        remove_ipynb_checkpoints()
-        estimate_pose()
-        # Generate human parse
-        remove_ipynb_checkpoints()
-        generate_human_segmentation_map()
-        generate_test_pairs_txt()
-        remove_ipynb_checkpoints()
-        generate_viton_hd()
-        for p in ["/content/fashion-eye-try-on/output/person_cloth.jpg", "/content/fashion-eye-try-on/output/person.jpg_cloth.jpg"]:
-            if os.path.exists(p):
-                result_img = Image.open(p)
-    except Exception as e:
-        print(e)
-        return
-    return result_img
-
+  result_img = person_img
+  # img.save(IMAGEPATH + "person.jpg")
+  # dress.save(CLOTHPATH + "cloth.jpg")
+
+  # txt = open("/content/VITON-HD/datasets/test_pairs.txt", "a")
+  # txt.write("person_img.jpg dress_img.jpg\n")
+  # txt.close()
+  # # result
+  # print(person_img.info, cloth_img.info)
+  # p_t1 = Thread(target=upload_remove_background_and_save_person_image, args=(person_img, ))
+  # c_t2 = Thread(target=upload_resize_generate_cloth_mask_and_move_to_viton_hd_test_inputs, args=(cloth_img, ))
+  # p_t1.start()
+  # c_t2.start()
+  # p_t1.join()
+  # c_t2.join()
+  # Estimate pose
+  try:
+    upload_resize_generate_cloth_mask_and_move_to_viton_hd_test_inputs(cloth_img)
+    upload_remove_background_and_save_person_image(person_img)
+    remove_ipynb_checkpoints()
+    estimate_pose()
+    # Generate human parse
+    remove_ipynb_checkpoints()
+    generate_human_segmentation_map()
+    generate_test_pairs_txt()
+    remove_ipynb_checkpoints()
+    generate_viton_hd()
+    for p in ["/content/fashion-eye-try-on/output/person_cloth.jpg", "/content/fashion-eye-try-on/output/person.jpg_cloth.jpg"]:
+        if os.path.exists(p):
+            result_img = Image.open(p)      
+  except Exception as e:
+    print(e)
+    return
+  return result_img
 
 # res = fashion_eye_tryon("", "")
 # res.show()
-gr.Interface(fn=fashion_eye_tryon,
-             inputs=[gr.Image(type="pil", label="Your image"),
-                     gr.Image(type="pil", label="Dress")],
+gr.Interface(fn=fashion_eye_tryon, 
+             inputs=[gr.Image(type = "pil", label="Your image"), gr.Image(type="pil", label="Dress")],
              outputs="image"
-             ).launch(debug=True, inbrowser=True, share=True)
+             ).launch(debug=True, inbrowser=True, share = True)
 
-# !pip freeze > /content/requirements_final.txt
+# !pip freeze > /content/requirements_final.txt