EndlessSora commited on
Commit
2c9f146
·
1 Parent(s): 10bdb89

update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -154,12 +154,12 @@ def generate_image(
154
  enable_anti_blur,
155
  model_version
156
  ):
157
- pipeline = prepare_pipeline(model_version=model_version, enable_realism=enable_realism, enable_anti_blur=enable_anti_blur)
 
158
 
159
- if seed == 0:
160
- seed = torch.seed() & 0xFFFFFFFF
161
 
162
- try:
163
  image = pipeline(
164
  id_image=input_image,
165
  prompt=prompt,
@@ -178,7 +178,7 @@ def generate_image(
178
  gr.Error(f"An error occurred: {e}")
179
  return gr.update()
180
 
181
- return gr.update(value = image, label=f"Generated image, seed = {seed}")
182
 
183
 
184
  def generate_examples(id_image, control_image, prompt_text, seed, enable_realism, enable_anti_blur, model_version):
 
154
  enable_anti_blur,
155
  model_version
156
  ):
157
+ try:
158
+ pipeline = prepare_pipeline(model_version=model_version, enable_realism=enable_realism, enable_anti_blur=enable_anti_blur)
159
 
160
+ if seed == 0:
161
+ seed = torch.seed() & 0xFFFFFFFF
162
 
 
163
  image = pipeline(
164
  id_image=input_image,
165
  prompt=prompt,
 
178
  gr.Error(f"An error occurred: {e}")
179
  return gr.update()
180
 
181
+ return gr.update(value=image, label=f"Generated image, seed = {seed}")
182
 
183
 
184
  def generate_examples(id_image, control_image, prompt_text, seed, enable_realism, enable_anti_blur, model_version):