test

app.py

@@ -6,6 +6,7 @@ import numpy as np
 import random
 import base64
 import requests
+import json
 
 
 def start_tryon(person_img, garment_img, seed, randomize_seed):
@@ -19,8 +20,24 @@ def start_tryon(person_img, garment_img, seed, randomize_seed):
     url = "http://" + os.environ['tryon_url']
     token = os.environ['token']
     print(url, token)    
-    
-    return person_img, seed
+    headers = {'Content-Type': 'application/json', 'token': token}
+    data = {
+        "clothImage": encoded_garment_img,
+        "humanImage": encoded_person_img,
+        "seed": seed
+    }
+
+    # response = requests.post(url, headers=headers, data=json.dumps(data))
+    # print("response code", response.status_code)
+    # if response.status_code == 200:
+    #     result = response.json()
+    #     result = base64.b64decode(result['images'][0])
+    #     result_np = np.frombuffer(result, np.uint8)
+    #     result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED)
+
+    result_img = cv2.imdecode(np.frombuffer(base64.b64decode(encoded_person_img), np.uint8), cv2.IMREAD_UNCHANGED)
+
+    return result_img, seed
 
 MAX_SEED = 999999
 

requirements.txt

@@ -5,4 +5,5 @@ torch
 transformers
 xformers
 opencv-python
-requests
+requests
+json