finalize content

Author: eaidova

notebooks/flex.2-image-generation/README.md

@@ -1,23 +1,24 @@
 # Image generation with universal control using Flex.2 and OpenVINO
 
+<div class="alert alert-block alert-danger"> <b>Important note:</b> This notebook requires python >= 3.11. Please make sure that your environment fulfill to this requirement before running it </div>
+
 Flex.2 is flexible text-to-image diffusion model based on Flux model architecture with built in support inpainting and universal control - model accepts pose, line, and depth inputs.
 
+<img src="https://github.com/user-attachments/assets/6a9ab66a-387a-4538-8625-2bb3a16072b5" width="1024"> 
+
 More details about model can be found in [model card](https://huggingface.co/ostris/Flex.2-preview).
 
 In this tutorial we consider how to convert and optimize Flex.2 model using OpenVINO.
 
 >**Note**: Some demonstrated models can require at least 32GB RAM for conversion and running.
 
-<img src="https://raw.githubusercontent.com/black-forest-labs/flux/main/assets/grid.jpg" width="1024"> 
-
 ### Notebook Contents
 
 In this demonstration, you will learn how to perform text-to-image generation using Flex.2 and OpenVINO. 
 
 Example of model work:
 
-**Input prompt**: *a tiny Yorkshire terrier astronaut hatching from an egg on the moon*
-![](https://github.com/user-attachments/assets/11733314-0b31-449c-9885-12ebf6365a58)
+![](https://github.com/user-attachments/assets/140685b7-2c5d-4cef-86fb-33df0849ec1a)
 
 The tutorial consists of the following steps:
 
@@ -33,4 +34,5 @@ The tutorial consists of the following steps:
 
 This is a self-contained example that relies solely on its own code.</br>
 We recommend running the notebook in a virtual environment. You only need a Jupyter server to start.
-For further details, please refer to [Installation Guide](../../README.md).
+For further details, please refer to [Installation Guide](../../README.md).
+<img referrerpolicy="no-referrer-when-downgrade" src="https://static.scarf.sh/a.png?x-pxid=5b5a4db0-7875-4bfb-bdbd-01698b5b1a77&file=../notebooks/flex.2-image-generation/README.md" />

notebooks/flex.2-image-generation/flex.2-image-generation.ipynb

@@ -7,21 +7,45 @@
    "source": [
     "# Image generation with universal control using Flex.2 and OpenVINO\n",
     "\n",
+    "<div class=\"alert alert-block alert-danger\"> <b>Important note:</b> This notebook requires python >= 3.10. Please make sure that your environment fulfill to this requirement before running it </div>\n",
+    "\n",
     "Flex.2 is flexible text-to-image diffusion model based on Flux model architecture with built in support inpainting and universal control - model accepts pose, line, and depth inputs.\n",
     "\n",
+    "![examples.png](https://github.com/user-attachments/assets/6a9ab66a-387a-4538-8625-2bb3a16072b5)\n",
+    "\n",
     "More details about model can be found in [model card](https://huggingface.co/ostris/Flex.2-preview).\n",
     "\n",
     "In this tutorial we consider how to convert and optimize Flex.2 model using OpenVINO.\n",
     "\n",
-    ">**Note**: Some demonstrated models can require at least 32GB RAM for conversion and running."
+    ">**Note**: Some demonstrated models can require at least 32GB RAM for conversion and running.\n",
+    "#### Table of contents:\n",
+    "\n",
+    "- [Prerequisites](#Prerequisites)\n",
+    "- [Convert model with OpenVINO](#Convert-model-with-OpenVINO)\n",
+    "    - [Convert model using Optimum Intel](#Convert-model-using-Optimum-Intel)\n",
+    "    - [Compress model weights](#Compress-model-weights)\n",
+    "- [Run OpenVINO model inference](#Run-OpenVINO-model-inference)\n",
+    "    - [Select inference device](#Select-inference-device)\n",
+    "- [Interactive demo](#Interactive-demo)\n",
+    "\n",
+    "\n",
+    "### Installation Instructions\n",
+    "\n",
+    "This is a self-contained example that relies solely on its own code.\n",
+    "\n",
+    "We recommend  running the notebook in a virtual environment. You only need a Jupyter server to start.\n",
+    "For details, please refer to [Installation Guide](https://github.com/openvinotoolkit/openvino_notebooks/blob/latest/README.md#-installation-guide).\n",
+    "\n",
+    "<img referrerpolicy=\"no-referrer-when-downgrade\" src=\"https://static.scarf.sh/a.png?x-pxid=5b5a4db0-7875-4bfb-bdbd-01698b5b1a77&file=../notebooks/flex.2-image-generation/flex.2-image-generation.ipynb\" />\n"
    ]
   },
   {
    "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Prerequisites"
+    "## Prerequisites\n",
+    "[back to top ⬆️](#Table-of-contents:)"
    ]
   },
   {
@@ -106,7 +130,9 @@
   {
    "cell_type": "code",
    "execution_count": 3,
-   "metadata": {},
+   "metadata": {
+   "test_replace": {"ostris/Flex.2-preview": "katuni4ka/tiny-random-flex.2-preview"}
+   },
    "outputs": [
     {
      "data": {
@@ -148,7 +174,7 @@
     "# Read more about telemetry collection at https://github.com/openvinotoolkit/openvino_notebooks?tab=readme-ov-file#-telemetry\n",
     "from notebook_utils import collect_telemetry\n",
     "\n",
-    "collect_telemetry(\"flex2-image-generation.ipynb\")\n",
+    "collect_telemetry(\"flex.2-image-generation.ipynb\")\n",
     "\n",
     "\n",
     "model_path = model_base_path / \"INT8\" if to_compress.value else model_base_path / \"FP16\"\n",
@@ -173,10 +199,12 @@
    "metadata": {},
    "source": [
     "## Run OpenVINO model inference\n",
+    "[back to top ⬆️](#Table-of-contents:)\n",
     "\n",
     "Flex.2 is based on Flux.1 model, but for enabling image control and inpainting capability, model uses own customized pipeline. `ov_flex2_helper.py` contains `OVFlex2Pipeline` class, pipeline adoption for usage with OpenVINO. It is based on Optimum Intel inference API and preserves functional features from original pipeline.\n",
     "\n",
     "### Select inference device\n",
+    "[back to top ⬆️](#Table-of-contents:)\n",
     "\n",
     "Select device from dropdown list for running inference using OpenVINO."
    ]
@@ -344,7 +372,8 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Interactive demo"
+    "## Interactive demo\n",
+    "[back to top ⬆️](#Table-of-contents:)"
    ]
   },
   {
@@ -387,6 +416,22 @@
    "pygments_lexer": "ipython3",
    "version": "3.11.4"
   },
+  "openvino_notebooks": {
+   "imageUrl": "https://github.com/user-attachments/assets/6a9ab66a-387a-4538-8625-2bb3a16072b5",
+   "tags": {
+    "categories": [
+     "Model Demos",
+     "AI Trends"
+    ],
+    "libraries": [],
+    "other": [
+     "Stable Diffusion"
+    ],
+    "tasks": [
+     "Image-to-Image"
+    ]
+   }
+  },
   "widgets": {
    "application/vnd.jupyter.widget-state+json": {
     "state": {