flow is an amazing image model

App/Generate/database/Vercel.py

@@ -47,6 +47,7 @@ class AsyncImageGenerator:
                         "steps": 28,
                         "width": 1024,
                         "height": 1536,
+                        "aspect_ratio": "9:16",
                         "prompt": payload,
                         # "sampler": "dpmpp_2m_sde_gpu",
                         # "scheduler": "karras",
@@ -61,7 +62,8 @@ class AsyncImageGenerator:
                     # "path": "models/playgroundai/playground-v2.5-1024px-aesthetic/versions/a45f82a1382bed5c7aeb861dac7c7d191b0fdf74d8d57c4a0e6ed7d4d0bf7d24",
                     # "path": "models/fofr/any-comfyui-workflow/versions/cd385285ba75685a040afbded7b79814a971f3febf46c5eab7c716e200c784e1",
                     # "path": "models/fofr/sd3-explorer/versions/a9f4aebd943ad7db13de8e34debea359d5578d08f128e968f9a36c3e9b0148d4",
-                    "path": "models/stability-ai/stable-diffusion-3",
+                    # "path": "models/stability-ai/stable-diffusion-3",
+                    "path": "models/black-forest-labs/flux-schnell",
                     # "path": "models/lucataco/proteus-v0.4/versions/34a427535a3c45552b94369280b823fcd0e5c9710e97af020bf445c033d4569e",
                     # "path": "models/lucataco/juggernaut-xl-v9/versions/bea09cf018e513cef0841719559ea86d2299e05448633ac8fe270b5d5cd6777e",
                 }
@@ -123,7 +125,9 @@ async def process_images(payloads):
                 result = await image_generator.generate_image(payload)
                 status = await image_generator.fetch_image_status(result["id"])
                 print(status["output"])
-                results.extend(status["output"])
+                if type(status["output"]) == list:
+                    results.extend(status["output"])
+                results.append(status["output"])
 
         for payload in payloads:
             tasks.append(payload)
@@ -139,7 +143,7 @@ async def process_images(payloads):
 # # # Example payloads
 # payloads = [
 #     """
-# Awsome cat
+# comicbook illustration artistic, beautiful Awsome cat
 #      """
 # ]