QHL067 commited on
Commit
436b13d
·
1 Parent(s): c47371b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -165,7 +165,7 @@ def infer(
165
  prompt_dict = {"prompt_1": prompt1, "prompt_2": prompt2}
166
  for key, value in prompt_dict.items():
167
  assert value is not None, f"{key} must not be None."
168
- if operation_mode != 'Interpolation':
169
  assert num_of_interpolation >= 5, "For linear interpolation, please sample at least five images."
170
  else:
171
  assert num_of_interpolation == 3, "For arithmetic, please sample three images."
 
165
  prompt_dict = {"prompt_1": prompt1, "prompt_2": prompt2}
166
  for key, value in prompt_dict.items():
167
  assert value is not None, f"{key} must not be None."
168
+ if operation_mode == 'Interpolation':
169
  assert num_of_interpolation >= 5, "For linear interpolation, please sample at least five images."
170
  else:
171
  assert num_of_interpolation == 3, "For arithmetic, please sample three images."