Update app.py

app.py

@@ -905,30 +905,29 @@ def generate_embed_style(prompt, learned_style, seed):
 
 def generate_image_from_prompt(text_in, style_in):
 
-	prompt = 'A campfire (oil on canvas)'
-	style_seed = 32
-	dict_styles = {'<gartic-phone>':'learned_embeds_gartic-phone.bin',
-	               '<hawaiian shirt>':'learned_embeds_hawaiian-shirt.bin',
-	               '<gp>': 'learned_embeds_phone01.bin',
-	               '<style-spdmn>':'learned_embeds_style-spdmn.bin',
-	               '<yvmqznrm>': 'learned_embedssd_yvmqznrm.bin'}
-
-	learn_embed = ['learned_embeds_gartic-phone.bin', 'learned_embeds_hawaiian-shirt_style.bin', 'learned_embeds_phone01_style.bin', 'learned_embeds_style-spdmn_style.bin', 'learned_embedssd_yvmqznrm_style.bin']
-	style = dict_styles # (learn_embed[0])
-	birb_embed = torch.load(learn_embed[0])
-	#birb_embed.keys(), dict_styles['<gartic-phone>'].shape
-
-
-	#style_embed = torch.load(dict_styles)
-	#birb_embed = torch.load('learned_embeds.bin')
-	#birb_embed.keys(), birb_embed['<birb-style>'].shape
-	generated_image = generate_embed_style(prompt,birb_embed, style_seed)
-	generate_loss_details = (generate_loss_style(prompt, birb_embed, style_seed))
-#generate_loss_style(prompt, style_embed, style_seed):
-
-#loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
-
-	return [generated_image,generate_loss_details]
+    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']
+    #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]
+
+    STYLE_SEEDS = [128, 64, 128, 64, 128]
+    
+    print(text_in)
+    print(style_in)
+    style_file = style_in + '_style.bin'
+    idx = STYLE_LIST.index(style_file)
+    print(style_file)
+    print(idx)  
+    
+    prompt = text_in 
+    
+    style_seed = STYLE_SEEDS[idx]
+    style_dict = torch.load(style_file)
+    style_embed = [v for v in style_dict.values()]
+
+    generated_image = embed_style(prompt, style_embed[0], style_seed)
+
+    loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
+
+    return [generated_image]
 
 
 # Define Interface