Spaces:

RashiAgarwal
/

GenerativeArt_StableDiffusion

Runtime error

RashiAgarwal commited on Oct 13, 2023

Commit

d41d6ef

1 Parent(s): 3b6f9cd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -153,7 +153,7 @@ def ref_loss(images,ref_image):
 def inference(prompt, style_index):
     styles = ['<midjourney-style>', '<hitokomoru-style>','<birb-style>','<summie-style>','<illustration-style>','<m-geo>','<buhu>']
-    embed = ['learned_embeds_m.bin','learned_embeds_h.bin', 'learned_embeds.bin',   'learned_embeds_s.bin','learned_embeds_i.bin','learned_embeds_mg.bin','learned_embeds_buhu']
     # Tokenize

 def inference(prompt, style_index):
     styles = ['<midjourney-style>', '<hitokomoru-style>','<birb-style>','<summie-style>','<illustration-style>','<m-geo>','<buhu>']
+    embed = ['learned_embeds_m.bin','learned_embeds_h.bin', 'learned_embeds.bin',   'learned_embeds_s.bin','learned_embeds_i.bin','learned_embeds_mg.bin','learned_embeds_buhu.bin']
     # Tokenize