AYYasaswini commited on
Commit
4969a8b
1 Parent(s): 28ee572

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- prompt = text_in
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)