more demo page tweaks

data/demo/index.template.html

@@ -6,6 +6,8 @@
 		<h1>VALL-E Demo</h1>
 		<p>${PREAMBLE}</p>
 		<table>
+			<thead>
+				<caption>LibriSpeech</caption>
 				<tr>
 					<th>Text</th>
 					<th>Prompt</th>
@@ -13,15 +15,22 @@
 					<th>Our VALL-E</th>
 					<th>Original VALL-E</th>
 					<th>YourTTS</th>
-			</tr>${LIBRISPEECH_SAMPLES}
+				</tr>
+			</thead>
+			<tbody>${LIBRISPEECH_SAMPLES}</tbody>
 		</table>
 		<table>
+			<thead>
+				<caption>Sampled Dataset</caption>
 				<tr>
 					<th>Text</th>
 					<th>Prompt</th>
 					<th>Ground Truth</th>
 					<th>Our VALL-E</th>
-			</tr>${DATASET_SAMPLES}
+				</tr>
+			</thead>
+			<tbody>${DATASET_SAMPLES}</tbody>
 		</table>
+		<p>Settings used: <pre>${SETTINGS}</pre></p>
 	</body>
 </html>

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 = {