New8893 commited on
Commit
f0d2f04
·
verified ·
1 Parent(s): 422adcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -23
app.py CHANGED
@@ -8,23 +8,6 @@ import requests
8
  import json
9
  import time
10
 
11
- def tryon(person_img, garment_img, seed, randomize_seed):
12
- """
13
- MOCK MODE: จำลองการทำ Try-On โดยแสดงภาพตัวอย่างแทนการเรียก API จริง
14
- """
15
- import cv2
16
- import os
17
-
18
- if person_img is None or garment_img is None:
19
- return None, None, "Please upload both person and garment image."
20
-
21
- # โหลดภาพตัวอย่างจากโฟลเดอร์ assets
22
- example_result_path = os.path.join("assets", "examples", "result1.png")
23
- result_img = cv2.imread(example_result_path)
24
- result_img = cv2.cvtColor(result_img, cv2.COLOR_BGR2RGB)
25
-
26
- return result_img, seed, "Mocked success"
27
- '''
28
  def tryon(person_img, garment_img, seed, randomize_seed):
29
  post_start_time = time.time()
30
  if person_img is None or garment_img is None:
@@ -41,17 +24,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
41
  token = os.environ['token']
42
  cookie = os.environ['Cookie']
43
  referer = os.environ['referer']
44
-
45
- os.environ['tryon_url'] = "127.0.0.1:8000/"
46
- os.environ['token'] = "your_token"
47
- os.environ['Cookie'] = "your_cookie"
48
- os.environ['referer'] = "your_referer"
49
  headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
50
  data = {
51
  "clothImage": encoded_garment_img,
52
  "humanImage": encoded_person_img,
53
  "seed": seed
54
- }'''
55
  try:
56
  response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
57
  # print("post response code", response.status_code)
 
8
  import json
9
  import time
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  def tryon(person_img, garment_img, seed, randomize_seed):
12
  post_start_time = time.time()
13
  if person_img is None or garment_img is None:
 
24
  token = os.environ['token']
25
  cookie = os.environ['Cookie']
26
  referer = os.environ['referer']
 
 
 
 
 
27
  headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
28
  data = {
29
  "clothImage": encoded_garment_img,
30
  "humanImage": encoded_person_img,
31
  "seed": seed
32
+ }
33
  try:
34
  response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
35
  # print("post response code", response.status_code)