Update api.py

rom1win · Oct 17, 2022 · 71d42bb · 71d42bb
1 parent 99013ba
commit 71d42bb
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/modules/api/api.py b/modules/api/api.py
@@ -23,24 +23,15 @@ def __init__(self, txt2img, img2img, run_extras, run_pnginfo):
         app.add_api_route("/v1/txt2img", self.text2imgapi, methods=["POST"])
 
     def text2imgapi(self, txt2imgreq: StableDiffusionProcessingAPI ):
-        print(txt2imgreq)
         p = StableDiffusionProcessingTxt2Img(**vars(txt2imgreq))
         p.sd_model = shared.sd_model
-        print(p)
         processed = process_images(p)
 
         b64images = []
         for i in processed.images:
             buffer = io.BytesIO()
             i.save(buffer, format="png")
             b64images.append(base64.b64encode(buffer.getvalue()))
-
-        response = {
-            "images": b64images,
-            "info": processed.js(),
-            "parameters": json.dumps(vars(txt2imgreq))
-        }
-
 
         return TextToImageResponse(images=b64images, parameters=json.dumps(vars(txt2imgreq)), info=json.dumps(processed.info))
 
@@ -57,4 +48,4 @@ def pnginfoendoint(self):
 
     def launch(self, server_name, port):
         app.include_router(self.router)
-        uvicorn.run(app, host=server_name, port=port)
+        uvicorn.run(app, host=server_name, port=port)