Merge pull request #18 from jhj0517/add-notebook

app.py

@@ -4,6 +4,13 @@ from modules.nllb_inference import NLLBInference
 import os
 from ui.htmls import *
 from modules.youtube_manager import get_ytmetas
+import argparse
+
+# Create the parser
+parser = argparse.ArgumentParser()
+parser.add_argument('--share', type=bool, default=False, nargs='?', const=True,
+                    help='Share value')
+args = parser.parse_args()
 
 
 def open_folder(folder_path):
@@ -20,7 +27,6 @@ def on_change_models(model_size):
     else:
         return gr.Checkbox.update(visible=True, value=False, label="Translate to English?", interactive=True)
 
-
 whisper_inf = WhisperInference()
 nllb_inf = NLLBInference()
 block = gr.Blocks(css=CSS).queue(api_open=False)
@@ -127,4 +133,7 @@ with block:
             btn_openfolder.click(fn=lambda: open_folder(os.path.join("outputs", "translations")), inputs=None, outputs=None)
 
 
-block.launch()
+if args.share:
+    block.launch(share=True)
+else:
+    block.launch()

notebook/whisper-webui.ipynb

@@ -0,0 +1,75 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "source": [
+        "**This notebook has been updated in [here](https://github.com/jhj0517/Whisper-WebUI.git) !**"
+      ],
+      "metadata": {
+        "id": "doKhBBXIfS21"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "#@title #Check GPU\n",
+        "#@markdown Some models may not function correctly on a CPU runtime. \n",
+        "\n",
+        "#@markdown so you should check your GPU setup before run.\n",
+        "!nvidia-smi"
+      ],
+      "metadata": {
+        "id": "23yZvUlagEsx",
+        "cellView": "form"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "kNbSbsctxahq",
+        "cellView": "form"
+      },
+      "outputs": [],
+      "source": [
+        "#@title #Installation\n",
+        "#@markdown This cell will install dependencies for Whisper-WebUI!\n",
+        "!git clone https://github.com/jhj0517/Whisper-WebUI.git\n",
+        "%cd Whisper-WebUI\n",
+        "!pip install -r requirements.txt"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "PQroYRRZzQiN",
+        "cellView": "form"
+      },
+      "outputs": [],
+      "source": [
+        "#@title #Run\n",
+        "#@markdown Once the installation is complete, you can use public URL that is displayed.\n",
+        "!python app.py --share"
+      ]
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": [],
+      "gpuType": "A100"
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    },
+    "accelerator": "GPU"
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}