gotta sleep

19a85813 · novelailab · f528ff54 · 19a85813
Commit 19a85813 authored Jul 30, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

main.py main.py +1 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -74,7 +74,7 @@ def generate(request: GenerationRequest):
            return ErrorOutput(error=output[1])
        images = model.sample(request)
-        for x in range(images):
+        for x in range(len(images)):
            image = simplejpeg.encode_jpeg(images[x], quality=95) 
            image = image.tobytes()
            #get base64 of image