test different art styles, change pipeline prompt

3d-generation-pipeline/.gitignore

@@ -3,4 +3,5 @@
 __pycache__
 images/
 models/
-logs/
+logs/
+notebooks/test_resources/

3d-generation-pipeline/notebooks/cloudflare_API_test.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "1dc6faae",
    "metadata": {},
    "outputs": [],
@@ -15,7 +15,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "b3107275",
    "metadata": {},
    "outputs": [],
@@ -44,17 +44,18 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Saved: output.jpg  (263282 bytes)\n",
-      "Saved: image9.jpg  (263282 bytes)\n"
+      "Saved: test_resources/resolution_test_1.jpg  (315728 bytes)\n"
      ]
     }
    ],
    "source": [
+    "# https://developers.cloudflare.com/workers-ai/models/flux-1-schnell/\n",
+    "\n",
     "MODEL = \"@cf/black-forest-labs/flux-1-schnell\"\n",
     "URL = f\"https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/{MODEL}\"\n",
     "\n",
     "payload = {\n",
-    "    \"prompt\": \"a slightly curved broadsword with a fancy golden crossguard\",\n",
+    "    \"prompt\": \"cyborg crocodile, realistic style, single object, front and side fully visible, plain neutral background, clear details, soft studio lighting, true-to-scale\",\n",
     "}\n",
     "\n",
     "headers = {\n",
@@ -72,7 +73,7 @@
     "\n",
     "img_bytes = base64.b64decode(b64)\n",
     "\n",
-    "out_path = \"output.jpg\"\n",
+    "out_path = \"test_resources/resolution_test_1.jpg\"\n",
     "with open(out_path, \"wb\") as f:\n",
     "    f.write(img_bytes)\n",
     "\n",

3d-generation-pipeline/start_pipeline.py

@@ -33,7 +33,7 @@ async def main():
 
     input_prompt = args.prompt
     print(f"Input prompt: {input_prompt}")
-    image_generation_prompt = input_prompt + ", single object, fully visible, plain neutral background, high-resolution, sharp details, soft studio lighting, front and side visible, realistic style, true-to-scale"
+    image_generation_prompt = input_prompt + ", realistic style, single object, front and side fully visible, plain neutral background, clear details, soft studio lighting, true-to-scale"
     
     pipeline_folder = Path(__file__).resolve().parent
     timestamp = get_timestamp()