Update app.py

app.py

@@ -133,19 +133,17 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
 
     print(nsfw_prediction)
 
-    image = images[0].convert("RGBA")
-    image = image.resize((width, height))
-    json_data = []
+    image = images[0].convert("RGBA").resize((width, height))
+    buffer_data = []
 
-    for x in range(image.width):
-        json_data.append([])
-        for y in range(image.height):
+    for y in range(height):
+        for x in range(width):
             r, g, b, a = image.getpixel((x, y))
-            json_data[x].append([r, g, b, a])
+            buffer_data.extend([r, g, b, a])
 
-    json_string = json.dumps(json_data)
+    buffer_json = json.dumps(buffer_data)
 
-    return image_paths, {item['label']: round(item['score'], 3) for item in nsfw_prediction}, json_string
+    return image_paths, {item['label']: round(item['score'], 3) for item in nsfw_prediction}, buffer_json
 
 def cloud():
     print("[CLOUD] | Space maintained.")