add button to open in colab!

ESRGAN_MANGA.ipynb

@@ -0,0 +1,72 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "9LQUHs0MNqsh"
+      },
+      "source": [
+        "# ESRGAN Upscaling With Custom Models\n",
+        "\n",
+        "<br>\n",
+        "\n",
+        "This space uses old ESRGAN architecture to upscale images, using models made by the community.\n",
+        "\n",
+        "Once the photo upscaled (*it can take a long time, this space only uses CPU*), click or tap the **download button** under the image to download it. **The preview image is not the upscaled one.**\n",
+        "\n",
+        "To run, just click or tap the play button below, and open the second Gradio link.\n",
+        "\n",
+        "<br>\n",
+        "\n",
+        "<a href='https://ko-fi.com/Y8Y7GVAAF' target='_blank' style='display:block;margin-bottom:5px'><img height='36' style='border:0px;height:36px;' src='https://storage.ko-fi.com/cdn/kofi1.png?v=3' border='0' alt='Buy Me a Coffee at ko-fi.com' /></a>"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "IuJNlXTDFNRi"
+      },
+      "source": [
+        "## Clone Hugging Face Spaces Repo and install requeriments and start Gradio server"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "KzHQxbDoEHVm",
+        "outputId": "a0aad94e-df09-4207-dca4-1845c5d5d59a"
+      },
+      "outputs": [],
+      "source": [
+        "from IPython.display import clear_output\n",
+        "!git lfs install\n",
+        "!git lfs clone \"https://huggingface.co/spaces/0x90e/ESRGAN-MANGA\"\n",
+        "%cd ESRGAN-MANGA\n",
+        "!pip install -r requirements.txt -q\n",
+        "!pip install gradio\n",
+        "clear_output()\n",
+        "!python app.py"
+      ]
+    }
+  ],
+  "metadata": {
+    "accelerator": "GPU",
+    "colab": {
+      "provenance": []
+    },
+    "gpuClass": "standard",
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}

app.py

@@ -19,6 +19,7 @@ css = '''
     .file-preview div div:nth-child(3) {
         text-align: right !important;
         padding: 0.5rem 0;
+        width: auto;
     }
 
     #preview_img {
@@ -45,6 +46,10 @@ css = '''
         color: white;
         background-color: gray;
     }
+
+    .colab_img {
+        margin: 10px 0;
+    }
 '''
 
 title = "ESRGAN Upscaling With Custom Models"
@@ -52,14 +57,14 @@ title = "ESRGAN Upscaling With Custom Models"
 with gr.Blocks(title=title, css=css) as demo:
     gr.Markdown(
     f"""
-    # {title}
-    This space uses old ESRGAN architecture to upscale images, using models made by the community.
+# {title}
+This space uses old ESRGAN architecture to upscale images, using models made by the community.
 
-    Once the photo upscaled (*it can take a long time, this space only uses CPU*), click or tap the **download button** under the image to download it. **The preview image is not the upscaled one**
-    
-    **Colab coming soon™**
+Once the photo upscaled (*it can take a long time, this space only uses CPU*), click or tap the **download button** under the image to download it. **The preview image is not the upscaled one**
     """)
 
+    gr.HTML(value="For faster upscaling: <a href='https://colab.research.google.com/drive/1QfOA6BBdL4NrUmx-9d-pjacxNfu81HQo#scrollTo=H7qo-6AWFbLH' target='_blank'><img class='colab_img' src='https://colab.research.google.com/assets/colab-badge.svg' alt='Open In Colab'></a>")
+
     gr.HTML(value="<a href='https://ko-fi.com/Y8Y7GVAAF' target='_blank' style='display:block;margin-bottom:5px'><img height='36' style='border:0px;height:36px;' src='https://storage.ko-fi.com/cdn/kofi1.png?v=3' border='0' alt='Buy Me a Coffee at ko-fi.com' /></a>")
 
     with gr.Box():
@@ -78,7 +83,7 @@ with gr.Blocks(title=title, css=css) as demo:
                 with gr.Row():
                     out_file = gr.File(interactive=False, show_label=False, elem_id="preview_file")
 
-    gr.HTML(value="<p><a href='https://upscale.wiki/wiki/Model_Database'>Model Database</a></p>")
+    gr.HTML(value="<p><a href='https://upscale.wiki/wiki/Model_Database'>Model Database</a></p><p><h2>Known issue:</h2> when downloading a photo for the second time in Colab, it doesn't work. If someone knows why, please create a thread in the discussion tab and I'll fix this issue.</p>")
         
     upscale_btn.click(process_image.inference, inputs=[input_image, upscale_size, upscale_type], outputs=[output_image, out_file])