Spaces:
Sleeping
Sleeping
File size: 2,740 Bytes
2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 a93c7a1 2228055 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
from io import BytesIO
import fal_client
import os
from PIL import Image
import requests
import time
from kolorsvton1024 import kolor_vton
UPLOAD_FOLDER = 'uploads'
DOWNLOAD_FOLDER = 'downloads'
save_directory = "downloads"
def user_dress_cat(user_pic, cloth_gen, index):
time_1 = time.time()
filename_1 = 'user_image.jpg'
filename_2 = 'cloth_image.jpg'
file_path_1 = os.path.join(UPLOAD_FOLDER, filename_1)
file_path_2 = os.path.join(UPLOAD_FOLDER, filename_2)
Image.fromarray(user_pic).save(file_path_1)
Image.fromarray(cloth_gen).save(file_path_2)
time_2 = time.time()
save_time = time_2 - time_1
print(f"save_time:{save_time}")
time_1 = time.time()
handler = fal_client.submit(
"fal-ai/cat-vton",
arguments={
"human_image_url": fal_client.upload_file(file_path_1),
"garment_image_url": fal_client.upload_file(file_path_2),
"cloth_type": "overall"
},
)
request_id = handler.request_id
result = fal_client.result("fal-ai/cat-vton", request_id)
time_2 = time.time()
cat_time = time_2 - time_1
print(f"cat_time:{cat_time}")
time_1 = time.time()
response = requests.get(result['image']['url'])
time_2 = time.time()
url_time = time_2 - time_1
print(f"url_time:{url_time}")
time_1 = time.time()
if response.status_code == 200:
filename = os.path.join(save_directory, f"cat-vton_{index}.png")
with open(filename, 'wb') as f:
f.write(response.content)
else:
print(f"Failed to download image from {result['image']['url']}")
time_2 = time.time()
downloads_time = time_2 - time_1
print(f"downloads_time:{downloads_time}")
return os.path.join(save_directory, f"cat-vton_{index}.png")
def user_dress_kolor(user_pic, cloth_gen, index):
time_1 = time.time()
filename_1 = 'user_image.jpg'
filename_2 = 'cloth_image.jpg'
file_path_1 = os.path.join(UPLOAD_FOLDER, filename_1)
file_path_2 = os.path.join(UPLOAD_FOLDER, filename_2)
Image.fromarray(user_pic).save(file_path_1)
Image.fromarray(cloth_gen).save(file_path_2)
time_2 = time.time()
save_time = time_2 - time_1
print(f"save_time:{save_time}")
time_1 = time.time()
kolor_vton(file_path_1, file_path_2, index)
time_2 = time.time()
cat_time = time_2 - time_1
print(f"kolor_time:{cat_time}")
def user_cloths(user_pic, cloth_gen):
user_cloth = []
for i in range(1, 4):
user_cloth.append(user_dress_cat(user_pic, cloth_gen, i))
for i in range(4, 7):
user_dress_kolor(user_pic, cloth_gen, i)
user_cloth.append(os.path.join(save_directory, f"kolor/kolor_{i}.png"))
return user_cloth
|