images_encoded

main.py

@@ -87,17 +87,20 @@ def generate(request: GenerationRequest):
             return ErrorOutput(error=output[1])
         
         images = model.sample(request)
+        images_encoded = []
         for x in range(len(images)):
             image = simplejpeg.encode_jpeg(images[x], quality=95) 
             #get base64 of image
             image = base64.b64encode(image).decode("ascii")
-            images[x] = image
+            images_encoded.append(image)
+
+        del images
 
         process_time = time.perf_counter() - t
         logger.info(f"Request took {process_time:0.3f} seconds")
         data = ""
         ptr = 0
-        for x in images:
+        for x in images_encoded:
             ptr += 1
             data += ("event: newImage\nid: {}\ndata:{}\n\n").format(ptr, x)
         return Response(content=data, media_type="text/event-stream")
@@ -128,15 +131,18 @@ def generate(request: GenerationRequest):
             return ErrorOutput(error=output[1])
         
         images = model.sample(request)
+        images_encoded = []
         for x in range(len(images)):
             image = simplejpeg.encode_jpeg(images[x], quality=95) 
             #get base64 of image
             image = base64.b64encode(image).decode("ascii")
-            images[x] = image
+            images_encoded.append(image)
+
+        del images
 
         process_time = time.perf_counter() - t
         logger.info(f"Request took {process_time:0.3f} seconds")
-        return GenerationOutput(output=images)
+        return GenerationOutput(output=images_encoded)
 
     except Exception as e:
         traceback.print_exc()