Omer Karisman commited on
Commit
197e3ee
1 Parent(s): 90a9d0b

Fix inputs

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -121,7 +121,8 @@ def generate(
121
  if style_image is not None:
122
  style_image = load_and_resize_image(style_image, resolution, resolution)
123
  else:
124
- raise ValueError("You must provide a style image")
 
125
 
126
  if identity_image_1 is not None:
127
  identity_image_1 = load_and_resize_image(identity_image_1, resolution, resolution)
 
121
  if style_image is not None:
122
  style_image = load_and_resize_image(style_image, resolution, resolution)
123
  else:
124
+ style_image = base_image
125
+ # raise ValueError("You must provide a style image")
126
 
127
  if identity_image_1 is not None:
128
  identity_image_1 = load_and_resize_image(identity_image_1, resolution, resolution)