Update app.py

app.py

@@ -50,7 +50,7 @@ def set_base_models():
         safety_checker=None
     )
     img2img.to("cuda")
-    return None, img2img
+    return txt2img, img2img
 
 with gr.Blocks(css=css) as demo:
     with gr.Column():