Update app.py

app.py

@@ -34,7 +34,7 @@ def generate_output(name, date_of_birth, image):
         audio.export(audio_path, format="wav")
         print("Archivo de audio generado:", audio_path)
     except Exception as e:
-        return None, f"No se pudo generar el audio: {str(e)}"   
+        return None, f"No se pudo generar el audio: {str(e)}"
 
     command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face {image} --audio audio.wav --outfile video.mp4 --nosmooth"
     process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
@@ -57,11 +57,14 @@ def error_message_fn(error_message):
 
 def get_radio_labels():
     image_paths = ["1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg"]
+    image_labels = ["Andréa", "Rocío", "Ivana", "Georgina", "Sandra", "Liliana"]
     image_choices = []
-    for image_path in image_paths:
+
+    for image_path, image_label in zip(image_paths, image_labels):
         image = Image.open(image_path)
         image.thumbnail((50, 50))
-        image_choices.append(image)
+        image_choices.append((image, image_label))
+
     return image_choices
 
 def select_image(image):
@@ -70,7 +73,7 @@ def select_image(image):
 iface = gr.Interface(
     fn=select_image,
     inputs=[
-        gr.inputs.Radio(label="Selecciona una imagen:", choices=get_radio_labels(), type="index"),
+        gr.inputs.Image(label="Selecciona una imagen:", type="pil", choices=get_radio_labels(), default=get_radio_labels()[0]),
         gr.inputs.Textbox(lines=1, label="Nombre", placeholder="Ingresa tu nombre"),
         gr.inputs.Textbox(lines=1, label="Fecha de Nacimiento", placeholder="DD/MM/AAAA")
     ],