AYYasaswini commited on
Commit
28ee572
1 Parent(s): 719784d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -923,11 +923,13 @@ def generate_image_from_prompt(text_in, style_in):
923
  style_dict = torch.load(style_file)
924
  style_embed = [v for v in style_dict.values()]
925
 
926
- generated_image = embed_style(prompt, style_embed[0], style_seed)
 
 
927
 
928
- loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
929
 
930
- return [generated_image]
931
 
932
 
933
  # Define Interface
 
923
  style_dict = torch.load(style_file)
924
  style_embed = [v for v in style_dict.values()]
925
 
926
+ generated_image = generate_embed_style(prompt,style_embed[0], style_seed)
927
+ generate_loss_details = (generate_loss_style(prompt, birb_embed, style_seed))
928
+ #generate_loss_style(prompt, style_embed, style_seed):
929
 
930
+ #loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
931
 
932
+ return [generated_image,generate_loss_details]
933
 
934
 
935
  # Define Interface