AYYasaswini commited on
Commit
0c23a88
·
verified ·
1 Parent(s): 5ca6299

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -23
app.py CHANGED
@@ -279,29 +279,28 @@ def generate_loss_based_image(text_embeddings, seed, max_length):
279
 
280
  def generate_image_from_prompt(text_in, style_in):
281
 
282
- 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']
283
- #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]
284
-
285
- STYLE_SEEDS = [128, 64, 128, 64, 128]
286
-
287
- print(text_in)
288
- print(style_in)
289
- style_file = style_in + '_style.bin'
290
- idx = STYLE_LIST.index(style_file)
291
- print(style_file)
292
- print(idx)
293
-
294
- prompt = text_in
295
-
296
- style_seed = STYLE_SEEDS[idx]
297
- style_dict = torch.load(style_file)
298
- style_embed = [v for v in style_dict.values()]
299
-
300
- generated_image = embed_style(prompt, style_embed[0], style_seed)
301
-
302
- loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
303
-
304
- return [generated_image, loss_generated_img]
305
 
306
 
307
  # Define Interface
 
279
 
280
  def generate_image_from_prompt(text_in, style_in):
281
 
282
+ prompt = 'A campfire (oil on canvas)'
283
+ style_seed = 32
284
+ dict_styles = {'<gartic-phone>':'learned_embeds_gartic-phone.bin',
285
+ '<hawaiian shirt>':'learned_embeds_hawaiian-shirt.bin',
286
+ '<gp>': 'learned_embeds_phone01.bin',
287
+ '<style-spdmn>':'learned_embeds_style-spdmn.bin',
288
+ '<yvmqznrm>': 'learned_embedssd_yvmqznrm.bin'}
289
+
290
+ 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']
291
+ style = dict_styles # (learn_embed[0])
292
+ birb_embed = torch.load(learn_embed[0])
293
+ #birb_embed.keys(), dict_styles['<gartic-phone>'].shape
294
+ #style_embed = torch.load(dict_styles)
295
+ #birb_embed = torch.load('learned_embeds.bin')
296
+ #birb_embed.keys(), birb_embed['<birb-style>'].shape
297
+ generated_image = generate_embed_style(prompt,birb_embed, style_seed)
298
+ generate_loss_details = (generate_loss_style(prompt, birb_embed, style_seed))
299
+ #generate_loss_style(prompt, style_embed, style_seed):
300
+
301
+ #loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
302
+
303
+ return [generated_image]
 
304
 
305
 
306
  # Define Interface