John6666 commited on
Commit
24fb9ce
·
verified ·
1 Parent(s): 90ff00e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2073,7 +2073,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
2073
 
2074
  # 4) Implementations: both share the same signature via clone_signature
2075
  @clone_signature(_signature_src)
2076
- def generate_image(*, **kwargs) -> Tuple[str, Optional[List[str]], Optional[str]]:
2077
  _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
2078
  argv = _argv_from_kwargs(kwargs)
2079
  last = ("COMPLETE", None, None)
@@ -2082,7 +2082,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
2082
  return last
2083
 
2084
  @clone_signature(_signature_src)
2085
- def generate_image_stream(*, **kwargs) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
2086
  _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
2087
  argv = _argv_from_kwargs(kwargs)
2088
  yield from _generate_image(argv)
 
2073
 
2074
  # 4) Implementations: both share the same signature via clone_signature
2075
  @clone_signature(_signature_src)
2076
+ def generate_image(**kwargs) -> Tuple[str, Optional[List[str]], Optional[str]]:
2077
  _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
2078
  argv = _argv_from_kwargs(kwargs)
2079
  last = ("COMPLETE", None, None)
 
2082
  return last
2083
 
2084
  @clone_signature(_signature_src)
2085
+ def generate_image_stream(**kwargs) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
2086
  _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
2087
  argv = _argv_from_kwargs(kwargs)
2088
  yield from _generate_image(argv)