Update app.py

app.py

@@ -279,29 +279,28 @@ def generate_loss_based_image(text_embeddings, seed, max_length):
 
 def generate_image_from_prompt(text_in, style_in):
 
-    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, loss_generated_img]
+	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]
     
 
 # Define Interface