Merge pull request #78 from jhj0517/add-colab

app.py

@@ -203,6 +203,7 @@ parser.add_argument('--server_port', type=int, default=None, help='Gradio server
 parser.add_argument('--username', type=str, default=None, help='Gradio authentication username')
 parser.add_argument('--password', type=str, default=None, help='Gradio authentication password')
 parser.add_argument('--theme', type=str, default=None, help='Gradio Blocks theme')
+parser.add_argument('--colab', type=bool, default=False, nargs='?', const=True, help='Is colab user or not')
 _args = parser.parse_args()
 
 if __name__ == "__main__":

modules/subtitle_manager.py

@@ -119,8 +119,11 @@ def get_serialized_vtt(dicts):
 
 
 def safe_filename(name):
+    from app import _args
     INVALID_FILENAME_CHARS = r'[<>:"/\\|?*\x00-\x1f]'
     safe_name = re.sub(INVALID_FILENAME_CHARS, '_', name)
+    if not _args.colab:
+        return safe_name
     # Truncate the filename if it exceeds the max_length (20)
     if len(safe_name) > 20:
         file_extension = safe_name.split('.')[-1]

notebook/whisper-webui.ipynb

@@ -36,8 +36,7 @@
         "!nvidia-smi"
       ],
       "metadata": {
-        "id": "23yZvUlagEsx",
-        "cellView": "form"
+        "id": "23yZvUlagEsx"
       },
       "execution_count": null,
       "outputs": []
@@ -85,8 +84,7 @@
         "#@markdown If you wonder how these arguments are used, you can see the [Wiki](https://github.com/jhj0517/Whisper-WebUI/wiki/Command-Line-Arguments)."
       ],
       "metadata": {
-        "id": "Qosz9BFlGui3",
-        "cellView": "form"
+        "id": "Qosz9BFlGui3"
       },
       "execution_count": null,
       "outputs": []
@@ -95,17 +93,16 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "PQroYRRZzQiN",
-        "cellView": "form"
+        "id": "PQroYRRZzQiN"
       },
       "outputs": [],
       "source": [
         "#@title #Run\n",
         "#@markdown Once the installation is complete, you can use public URL that is displayed.\n",
         "if 'arguments' in locals():\n",
-        "  !python app.py --share{arguments}\n",
+        "  !python app.py --share --colab{arguments}\n",
         "else:\n",
-        "    !python app.py --share"
+        "    !python app.py --share --colab"
       ]
     }
   ],