sample inputs pull from examples first

This commit is contained in:
aliabd 2021-12-23 17:23:47 +04:00
parent 419598c87d
commit 24405444f1

View File

@ -244,7 +244,10 @@ def api_docs():
output_types_doc, output_types = get_types(outputs, "output")
input_names = [type(inp).__name__ for inp in app.interface.input_components]
output_names = [type(out).__name__ for out in app.interface.output_components]
sample_inputs = [inp.generate_sample() for inp in app.interface.input_components]
if app.interface.examples is not None:
sample_inputs = app.interface.examples[0]
else:
sample_inputs = [inp.generate_sample() for inp in app.interface.input_components]
docs = {
"inputs": input_names,
"outputs": output_names,