infnapitoggle / handlemodelradio.py
charliebaby2023's picture
Update handlemodelradio.py
bb4a6ad verified
raw
history blame
430 Bytes
from config import howManyModelsToUse,num_models
def extend_choices(choices):
return choices[:num_models] + (num_models - len(choices[:num_models])) * ['NA']
def update_imgbox(choices):
choices_plus = extend_choices(choices[:num_models])
return [gr.Image(None, label=m, visible=(m!='NA')) for m in choices_plus]
def random_choices():
import random
random.seed()
return random.choices(models, k=num_models)