Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import os
-import subprocess
 from gtts import gTTS
 from pydub import AudioSegment
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
@@ -10,10 +9,7 @@ generator = pipeline('text-generation', model="checkpoints")
 tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
 os.environ["TOKENIZERS_PARALLELISM"] = "true"
 
-def generate_output(name, date_of_birth, image_index):
-    choices = ["1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg"]
-    image = choices[image_index]
-    
+def generate_output(name, date_of_birth, image):
     prompt = f"Tu carta astral de hoy {name} es:"
     input_tokens = tokenizer.encode(prompt, add_special_tokens=False)
     input_text = tokenizer.decode(input_tokens)
@@ -61,19 +57,12 @@ def get_thumbnail(image_path, width):
 choices = ["1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg"]
 thumbnails = [get_thumbnail(image, 50) for image in choices]
 
-def get_slider_labels():
-    labels = []
-    for i in range(len(choices)):
-        label = f'<img src="data:image/png;base64,{gr.interface.utils.to_base64_string(thumbnails[i])}" width="50" height="50">'
-        labels.append(label)
-    return labels
-
 iface = gr.Interface(
     fn=generate_output,
     inputs=[
         gr.inputs.Textbox(lines=1, label="Nombre", placeholder="Ingresa tu nombre"),
         gr.inputs.Textbox(lines=1, label="Fecha de Nacimiento", placeholder="DD/MM/AAAA"),
-        gr.inputs.Slider(minimum=0, maximum=5, step=1, label="Selecciona una imagen:", labels=get_slider_labels())
+        gr.inputs.Image(label="Selecciona una imagen:", image_mode="RGB", choices=choices, thumbnails=thumbnails)
     ],
     outputs=[
         gr.outputs.Video(label="Respuesta de Andrea (un minuto aproximadamente)").style(width=256),