Merge remote-tracking branch 'origin/main'

Styp · Sep 17, 2022 · 2c2b252 · 2c2b252
2 parents 6899a25 + 5367dc5
commit 2c2b252
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/backend/serve.py b/backend/serve.py
@@ -28,14 +28,14 @@ def root():
 
 
 @app.post("/imageFromPromt")
-def image_from_promt(payload: PromptPayload) -> StreamingResponse:
+def image_from_promt(payload):
     consumer = StableDiffusionConsumer()
     image_bytes = consumer.fetch_image(payload.prompt)
 
     return StreamingResponse(consumer.parse_to_bytesio(image_bytes), media_type="image/png")
 
 @app.post("/base64FromPrompt")
-def base64_from_prompt(payload: PromptPayload) -> JSONResponse:
+def base64_from_prompt(payload):
     consumer = StableDiffusionConsumer()
     img_json = {
         "image": consumer.fetch_image(payload.prompt)
@@ -44,7 +44,7 @@ def base64_from_prompt(payload: PromptPayload) -> JSONResponse:
     return JSONResponse(content=img_json)
 
 @app.post("/promptsFromText")
-def prompts_from_text(text_payload: TextPayload) -> [Dict]:
+def prompts_from_text(text_payload):
     prompt_extractor = PromptExtractor()
 
     prompts = prompt_extractor.extract_paragraphs_with_prompts(text_payload.text)