mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-09 00:09:55 +00:00
Kasisnu
82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
import json
|
|
import time
|
|
import requests
|
|
|
|
# TEST_INPUT_FILE = 'test_image.jpg'
|
|
TEST_INPUT_FILE = 'test_image_10pc.jpg'
|
|
|
|
# url = 'http://b3104:8005/preview/'
|
|
# url = "http://207.189.112.61:31605/preview"
|
|
url = 'http://localhost:8000/preview/'
|
|
|
|
# url = 'https://funnel.tailce84f.ts.net/preview/'
|
|
style_to_filename = {
|
|
"anime_2_5d": "1_2.5d_anime_model.json",
|
|
"anime_2d_flat": "2_2d_flat_anime_model.json",
|
|
"cartoon_3d": "3_3d_cartoon_style.json",
|
|
"comic_book": "4_comic_book_model.json",
|
|
"anime_ghibli": "5_ghibli_anime_model.json",
|
|
"ink_punk": "6_ink_punk.json",
|
|
"ink_splash": "7_ink_splash.json",
|
|
"ink_bw_style": "8_ink_w_and_b_style.json",
|
|
"jojo_style": "9_jojo_style.json",
|
|
"paper_origami": "10_paper_origami.json",
|
|
"pixel_art": "11_pixel_art.json",
|
|
"pop_art": "12_pop_art.json",
|
|
"realistic_1": "13_realistic_1.json",
|
|
"realistic_2": "14_realistic_2.json",
|
|
"anime_retro_neon": "15_retro_neon_anime_90.json",
|
|
"anime_standard": "16_standard_anime_model.json",
|
|
"hr_giger": "17_hr_giger.json",
|
|
"simpsons": "18_simpsons.json",
|
|
"carnage": "19_carnage.json",
|
|
"pastel_cute_anime": "20_pastel_cute_anime.json",
|
|
"bloom_lighting": "21_bloom_lighting.json",
|
|
"25d_horror": "22_25D_Horror.json",
|
|
"creepy": "23_creepy.json",
|
|
"creepy_vhs": "24_creepy_vhs.json",
|
|
"trail_cam_footage": "25_trail_cam_footage.json",
|
|
"old_black_white_movie": "26_old_black_white_movie.json",
|
|
"horror_noir_black_white": "27_horror_noir_black_white.json",
|
|
"techno_noir_black_white": "28_techno_noir_black_white.json",
|
|
"black_white_20s": "29_black_white_20s.json",
|
|
"cyberpunk_anime": "30_cyberpunk_anime.json",
|
|
"dragonball": "31_dragonball.json",
|
|
"realistic_matrix": "32_realistic_matrix.json",
|
|
"realistic_cyberpunk": "33_realistic_cyberpunk.json",
|
|
"dreamer": "34_dreamer.json",
|
|
}
|
|
|
|
all_styles = list(style_to_filename.keys())
|
|
|
|
import random
|
|
random.shuffle(all_styles)
|
|
|
|
styles = list(all_styles)[:]
|
|
# random.shuffle(styles)
|
|
|
|
test_style_list = styles + styles
|
|
# for style, filename in list(style_to_filename.items())[:2]:
|
|
for style in all_styles:
|
|
for i in range(1):
|
|
# print(f"Testing style: {style}")
|
|
payload = {"style": style, "positive_prompt": "test pos in", "negative_prompt": "test neg in"}
|
|
with open(TEST_INPUT_FILE , 'rb') as input_file:
|
|
# files = {'input_file': ('input_file', input_file, 'image/jpeg')}
|
|
files = {'input_file': input_file}
|
|
# write output file suffixed with timestamp
|
|
timestamp = time.strftime('%Y%m%d%H%M%S')
|
|
output_file_name = f'outputs/output_image_{style}_{timestamp}.jpeg'
|
|
execution_start_time = time.perf_counter()
|
|
with requests.post(url, data={'request': json.dumps(payload)}, files=files) as response:
|
|
if response.status_code == 200:
|
|
with open(output_file_name, 'wb') as f:
|
|
f.write(response.content)
|
|
else:
|
|
# Handle request error
|
|
print(f"Request failed: {response.status_code}")
|
|
print(response.text)
|
|
current_time = time.perf_counter()
|
|
execution_time = current_time - execution_start_time
|
|
print('{} generated {} in {:.2f} seconds'.format(style, output_file_name, execution_time))
|