Shaamallow commited on
Commit
0773e7d
·
1 Parent(s): a18aa9e

fix example launching before loading image

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -131,6 +131,9 @@ def generate(
131
  ) -> PIL.Image.Image:
132
  torch.manual_seed(seed)
133
 
 
 
 
134
  if not use_negative_prompt:
135
  negative_prompt = None # type: ignore
136
  if not use_prompt_2:
 
131
  ) -> PIL.Image.Image:
132
  torch.manual_seed(seed)
133
 
134
+ if style_image_value is None:
135
+ gr.Error("Please provide a style image")
136
+
137
  if not use_negative_prompt:
138
  negative_prompt = None # type: ignore
139
  if not use_prompt_2: