Spaces:
Runtime error
Runtime error
import gradio as gr | |
from naamkaran.generate import generate_names | |
def gen_names(start_letter, end_letter, how_many, | |
max_length, gender, temperature): | |
""" | |
Generates names for the given inputs. | |
""" | |
if len(end_letter) > 1: | |
end_letter = end_letter[0] | |
if len(start_letter) > 1: | |
start_letter = start_letter[0] | |
if len(start_letter) < 1: | |
start_letter = "a" | |
if len(end_letter) < 1: | |
end_letter = None | |
if int(max_length) < 1: | |
max_length = 5 | |
if int(how_many) < 1: | |
how_many = 1 | |
names = generate_names(start_letter, end_letter, int(how_many), int(max_length), gender, temperature) | |
return ", ".join(names) | |
iface = gr.Interface( | |
fn=gen_names, | |
inputs=[gr.components.Text(value="A", label="Starting letter"), | |
gr.components.Text(value="e", label="Ending letter"), | |
gr.components.Number(value=5, label="How many to generate"), | |
gr.components.Number(value=5, label="Max length of the name"), | |
gr.components.Radio(["M", "F"], label="Gender", value="F"), | |
gr.components.Slider(0.1, 1, step=0.1, label='Temperature', value=0.5)], | |
outputs=gr.components.Textbox(lines=10, label="Names"), | |
title="Naamkaran", | |
description="Generate names for the given below inputs", | |
article="Naamkaran is a library to generate random names.", | |
allow_flagging="never", | |
examples=[ | |
["a", "n", 1, 5, "M", 0.5], | |
["a", "n", 1, 5, "F", 0.5], | |
] | |
) | |
if __name__ == "__main__": | |
iface.launch() | |