Update src/covergen.py

src/covergen.py

@@ -1,4 +1,4 @@
-import os
+import os, sys
 import json
 import shutil
 import urllib.request
@@ -16,13 +16,23 @@ BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 rvc_models_dir = os.path.join(BASE_DIR, 'rvc_models')
 output_dir = os.path.join(BASE_DIR, 'song_output')
 
+
+warning = sys.argv[1]
+
+if warning == 'True':
+    warning = True
+elif warning == 'False':
+    warning = False
+
+
 if __name__ == '__main__':
     voice_models = ignore_files(rvc_models_dir)
 
     with gr.Blocks(title='CoverGen Lite - Politrees (v0.2)', theme=gr.themes.Soft(primary_hue="green", secondary_hue="green", neutral_hue="neutral", spacing_size="sm", radius_size="lg")) as app:
         
-        with gr.Column(variant='panel'):
-            gr.HTML("<center><h2>This space is running too slow due to a weak server, so I made a Google Colab notebook to work faster with this interface: <a href='https://colab.research.google.com/drive/1HzuPgICRrjqUWQWb5Zn-l07m099-n-Nr'>Google Colab Notebook</a>.</h2></center>")
+        if warning:
+            with gr.Column(variant='panel'):
+                gr.HTML("<center><h2>This space is running too slow due to a weak server, so I made a Google Colab notebook to work faster with this interface: <a href='https://colab.research.google.com/drive/1HzuPgICRrjqUWQWb5Zn-l07m099-n-Nr'>Google Colab Notebook</a>.</h2></center>")
         
         with gr.Tab("Welcome/Contacts"):
             gr.HTML("<center><h1>Welcome to CoverGen Lite - Politrees (v0.2)</h1></center>")