From 3e5ca3a2010de9d4b175a9682bef89d391949b88 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Sun, 21 Jul 2024 19:31:13 -0500
Subject: [PATCH] more demo page tweaks

---
 data/demo/index.template.html | 37 ++++++++++++++++++++++-------------
 vall_e/demo.py                | 13 +++++++++++-
 2 files changed, 35 insertions(+), 15 deletions(-)

diff --git a/data/demo/index.template.html b/data/demo/index.template.html
index 0c0224e..0773a94 100644
--- a/data/demo/index.template.html
+++ b/data/demo/index.template.html
@@ -6,22 +6,31 @@
 		<h1>VALL-E Demo</h1>
 		<p>${PREAMBLE}</p>
 		<table>
-			<tr>
-				<th>Text</th>
-				<th>Prompt</th>
-				<th>Ground Truth</th>
-				<th>Our VALL-E</th>
-				<th>Original VALL-E</th>
-				<th>YourTTS</th>
-			</tr>${LIBRISPEECH_SAMPLES}
+			<thead>
+				<caption>LibriSpeech</caption>
+				<tr>
+					<th>Text</th>
+					<th>Prompt</th>
+					<th>Ground Truth</th>
+					<th>Our VALL-E</th>
+					<th>Original VALL-E</th>
+					<th>YourTTS</th>
+				</tr>
+			</thead>
+			<tbody>${LIBRISPEECH_SAMPLES}</tbody>
 		</table>
 		<table>
-			<tr>
-				<th>Text</th>
-				<th>Prompt</th>
-				<th>Ground Truth</th>
-				<th>Our VALL-E</th>
-			</tr>${DATASET_SAMPLES}
+			<thead>
+				<caption>Sampled Dataset</caption>
+				<tr>
+					<th>Text</th>
+					<th>Prompt</th>
+					<th>Ground Truth</th>
+					<th>Our VALL-E</th>
+				</tr>
+			</thead>
+			<tbody>${DATASET_SAMPLES}</tbody>
 		</table>
+		<p>Settings used: <pre>${SETTINGS}</pre></p>
 	</body>
 </html>
\ No newline at end of file
diff --git a/vall_e/demo.py b/vall_e/demo.py
index c13b019..46f201b 100644
--- a/vall_e/demo.py
+++ b/vall_e/demo.py
@@ -53,7 +53,7 @@ def main():
 	parser.add_argument("--nar-temp", type=float, default=0.0)
 	parser.add_argument("--min-ar-temp", type=float, default=-1.0)
 	parser.add_argument("--min-nar-temp", type=float, default=-1.0)
-	parser.add_argument("--input-prompt-length", type=float, default=3.0)
+	parser.add_argument("--input-prompt-length", type=float, default=0.0)
 
 	parser.add_argument("--top-p", type=float, default=1.0)
 	parser.add_argument("--top-k", type=int, default=16)
@@ -89,6 +89,17 @@ def main():
 
 	# replace values in our template
 	html = html.replace(r"${PREAMBLE}", args.preamble )
+	html = html.replace(r"${SETTINGS}", str(dict(
+		input_prompt_length=args.input_prompt_length,
+		max_ar_steps=args.max_ar_steps, max_nar_levels=args.max_nar_levels,
+		ar_temp=args.ar_temp, nar_temp=args.nar_temp,
+		min_ar_temp=args.min_ar_temp, min_nar_temp=args.min_nar_temp,
+		top_p=args.top_p, top_k=args.top_k,
+		repetition_penalty=args.repetition_penalty, repetition_penalty_decay=args.repetition_penalty_decay,
+		length_penalty=args.length_penalty,
+		beam_width=args.beam_width,
+		mirostat_tau=args.mirostat_tau, mirostat_eta=args.mirostat_eta,
+	)) )
 
 	# pull from provided samples
 	samples_dirs = {