From 9f76226ba6bace7a0393c6722fc8bebdd8bbc06f Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 28 Jan 2023 08:38:48 -0600
Subject: [PATCH] use correct generator

---
 api/onnx_web/serve.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index dd5a0211..ca2c0725 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -24,6 +24,7 @@ from typing import Tuple
 
 from .chain import (
     correct_gfpgan,
+    generate_txt2img,
     upscale_outpaint,
     upscale_resrgan,
     upscale_stable_diffusion,
@@ -544,7 +545,7 @@ def chain():
     params, size = pipeline_from_request()
 
     example = ChainPipeline(stages=[
-        (source_txt2img, StageParams(), None),
+        (generate_txt2img, StageParams(), None),
         (upscale_outpaint, StageParams(outscale=4), {
             'expand': Border(256, 256, 256, 256),
         }),