Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
283 |
-
|
284 |
-
|
285 |
-
|
286 |
-
|
287 |
-
|
288 |
-
|
289 |
-
|
290 |
-
|
291 |
-
|
292 |
-
|
293 |
-
|
294 |
-
|
295 |
-
|
296 |
-
|
297 |
-
|
298 |
-
|
299 |
-
|
300 |
-
|
301 |
-
|
302 |
-
|
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
|