Update app.py

app.py

@@ -19,6 +19,13 @@ from assets.discord_presence import rich_presence
 
 rich_presence()
 
+try:
+    from tensorflow.python.client import device_lib
+
+    gpus = [device.name for device in device_lib.list_local_devices() if device.device_type == "GPU"]
+except ImportError:
+    gpus = []
+
 with gr.Blocks(theme="ParityError/Interstellar", title="Applio") as Applio:
     gr.Markdown("# Applio")
     gr.Markdown(
@@ -34,8 +41,9 @@ with gr.Blocks(theme="ParityError/Interstellar", title="Applio") as Applio:
     with gr.Tab(i18n("Inference")):
         inference_tab()
 
-    with gr.Tab(i18n("Train")):
-        train_tab()
+    if gpus:
+        with gr.Tab(i18n("Train")):
+            train_tab()
 
     with gr.Tab(i18n("TTS")):
         tts_tab()
@@ -50,9 +58,5 @@ with gr.Blocks(theme="ParityError/Interstellar", title="Applio") as Applio:
         report_tab()
 
 if __name__ == "__main__":
-    Applio.launch(
-        favicon_path="assets/ICON.ico",
-        share="--share" in sys.argv,
-        inbrowser="--open" in sys.argv,
-        server_port=6969,
-    )
+    Applio.launch()
+    # share=True