Commit d5fcec8b authored by gd1551's avatar gd1551 Committed by GitHub

add uc to image metadata

parent 4f79dd7a
...@@ -139,7 +139,7 @@ def generate(request: GenerationRequest, authorized: bool = Depends(verify_token ...@@ -139,7 +139,7 @@ def generate(request: GenerationRequest, authorized: bool = Depends(verify_token
if seed is not None: if seed is not None:
request.seed = seed request.seed = seed
seed += 1 seed += 1
comment = json.dumps({"steps":request.steps,"sampler":request.sampler,"seed":request.seed,"strength":request.strength,"noise":request.noise,"scale":request.scale}) comment = json.dumps({"steps":request.steps,"sampler":request.sampler,"seed":request.seed,"strength":request.strength,"noise":request.noise,"scale":request.scale,"uc":request.uc})
metadata = PngInfo() metadata = PngInfo()
metadata.add_text("Title", "AI generated image") metadata.add_text("Title", "AI generated image")
metadata.add_text("Description", request.prompt) metadata.add_text("Description", request.prompt)
...@@ -251,4 +251,4 @@ def generate_text(request: TextRequest, authorized: bool = Depends(verify_token) ...@@ -251,4 +251,4 @@ def generate_text(request: TextRequest, authorized: bool = Depends(verify_token)
return ErrorOutput(error=str(e)) return ErrorOutput(error=str(e))
if __name__ == "__main__": if __name__ == "__main__":
uvicorn.run("main:app", host="0.0.0.0", port=80, log_level="info") uvicorn.run("main:app", host="0.0.0.0", port=80, log_level="info")
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment