Skip to content

Commit

Permalink
fix: return the output of "process_output_image" and access jobId cor…
Browse files Browse the repository at this point in the history
…rectly (#11)
  • Loading branch information
TimPietrusky authored Nov 17, 2023
1 parent b7101fc commit dc655ea
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rp_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,7 @@ def handler(job):
return {"error": f"Error waiting for image generation: {str(e)}"}

# Get the generated image and return it as URL in an AWS bucket or as base64
process_output_images(history[prompt_id].get("outputs"), job[id])
return process_output_images(history[prompt_id].get("outputs"), job["id"])


# Start the handler only if this script is run directly
Expand Down

0 comments on commit dc655ea

Please sign in to comment.