Editar um código existente em Python (Adicionar um dropdown e puxar um valor) [Orçamento de R$50-60]

Temos um código completo em Gradio que busca um modelo de IA e gera uma imagem. Acontece que queremos, via dropdown, adicionar mais modelos de IA. Quero que o valor do dropdown seja retornado mas não conseguimos fazer isso (parece muito fácil mas tem nos estressado bastante).

Código base: https://huggingface.co/spaces/hysts/sd-xl

basicamente: adicionar um dropdown e puxar o valor do dropdown na váriavel selected_model: selected_model = "valor do dropdown" # aqui precisa ir o valor do dropdown

if torch.cuda.is_available():
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
# Substitua o valor dinâmico antes de chamar from_pretrained
pipe_model_name = f"{selected_model}"
pipe = DiffusionPipeline.from_pretrained(
pipe_model_name,
vae=vae,
torch_dtype=torch.float16,
use_safetensors=True,
)

O código completo segue em anexo.