From bfc5e1d723faef9b6f13c42f10b9a35ec6087d5b Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Tue, 5 Nov 2024 22:30:49 -0600
Subject: [PATCH] agony

---
 README.md       |  2 ++
 vall_e/webui.py | 22 ++++++++++++++--------
 2 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index becf274..02b84b0 100755
--- a/README.md
+++ b/README.md
@@ -6,6 +6,8 @@
 
 An unofficial PyTorch implementation of [VALL-E](https://vall-e-demo.ecker.tech/), utilizing the [EnCodec](https://github.com/facebookresearch/encodec) encoder/decoder.
 
+A demo is available on HuggingFace [here](https://huggingface.co/spaces/ecker/vall-e).
+
 ## Requirements
 
 Besides a working PyTorch environment, the only hard requirement is [`espeak-ng`](https://github.com/espeak-ng/espeak-ng/) for phonemizing text:
diff --git a/vall_e/webui.py b/vall_e/webui.py
index 3b0b1ce..53aa410 100644
--- a/vall_e/webui.py
+++ b/vall_e/webui.py
@@ -1,5 +1,11 @@
-import os
 import sys
+import os
+
+argv = os.environ.get('VALLE_ARGS', None)
+
+if argv:
+	sys.argv = sys.argv + argv.split(" ")
+
 import re
 import math
 import argparse
@@ -29,7 +35,7 @@ try:
 
 	USING_SPACES = True
 	spaces_zerogpu_decorator = spaces.GPU
-except ImportError:
+except Exception as e:
 	USING_SPACES = False
 	def spaces_zerogpu_decorator(func):
 		return func
@@ -556,11 +562,11 @@ with ui:
 def start( lock=True ):
 	setup_logging()
 
-	ui.queue(max_size=8)
-	ui.launch(share=args.share, server_name=args.listen_host, server_port=args.listen_port, prevent_thread_lock=not lock)
+	if not USING_SPACES:
+		ui.queue(max_size=8)
+		ui.launch(share=args.share, server_name=args.listen_host, server_port=args.listen_port, prevent_thread_lock=not lock)
+	else:
+		ui.queue().launch()
 
 if __name__ == "__main__":
-	if not USING_SPACES:
-		start()
-	else:
-		ui.queue().launch()
\ No newline at end of file
+	start()
\ No newline at end of file