Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -904,7 +904,7 @@ def generate_embed_style(prompt, learned_style, seed):
|
|
904 |
|
905 |
|
906 |
def generate_image_from_prompt(text_in, style_in):
|
907 |
-
|
908 |
STYLE_LIST = ['learned_embeds_gartic-phone_style.bin', 'learned_embeds_hawaiian-shirt_style.bin', 'learned_embeds_phone01_style.bin', 'learned_embeds_style-spdmn_style.bin', 'learned_embedssd_yvmqznrm_style.bin']
|
909 |
#learned_embeds = [learned_embeds_gartic-phone.bin,learned_embeds_libraryhawaiian-shirt.bin, learned_embeds_phone0.bin1,learned_embeds_style-spdmn.bin,learned_embedssd_yvmqznrm.bin]
|
910 |
|
@@ -917,7 +917,7 @@ def generate_image_from_prompt(text_in, style_in):
|
|
917 |
print(style_file)
|
918 |
print(idx)
|
919 |
|
920 |
-
|
921 |
|
922 |
style_seed = STYLE_SEEDS[idx]
|
923 |
style_dict = torch.load(style_file)
|
|
|
904 |
|
905 |
|
906 |
def generate_image_from_prompt(text_in, style_in):
|
907 |
+
prompt = text_in
|
908 |
STYLE_LIST = ['learned_embeds_gartic-phone_style.bin', 'learned_embeds_hawaiian-shirt_style.bin', 'learned_embeds_phone01_style.bin', 'learned_embeds_style-spdmn_style.bin', 'learned_embedssd_yvmqznrm_style.bin']
|
909 |
#learned_embeds = [learned_embeds_gartic-phone.bin,learned_embeds_libraryhawaiian-shirt.bin, learned_embeds_phone0.bin1,learned_embeds_style-spdmn.bin,learned_embedssd_yvmqznrm.bin]
|
910 |
|
|
|
917 |
print(style_file)
|
918 |
print(idx)
|
919 |
|
920 |
+
|
921 |
|
922 |
style_seed = STYLE_SEEDS[idx]
|
923 |
style_dict = torch.load(style_file)
|