Update app.py

app.py

@@ -1,11 +1,15 @@
 import gradio as gr
 import os
+import subprocess
 from gtts import gTTS
 from pydub import AudioSegment
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
 from PIL import Image
-from io import BytesIO
-import base64
+
+def get_thumbnail(image_path, width):
+    image = Image.open(image_path)
+    image.thumbnail((width, width))
+    return image
 
 generator = pipeline('text-generation', model="checkpoints")
 tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
@@ -50,25 +54,16 @@ def error_message_fn(error_message):
         return gr.outputs.Textbox(text=error_message, placeholder="Error")
     else:
         return None
-
-def get_image_choices():
-    image_paths = ["1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg"]
-    image_choices = []
-    for image_path in image_paths:
-        image = Image.open(image_path)
-        image.thumbnail((50, 50))
-        image_choices.append((image_path, image))
-    return image_choices
-
-def select_image(image):
-    return image
+        
+choices = ["1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg"]
+thumbnails = [get_thumbnail(image, 50) for image in choices]
 
 iface = gr.Interface(
-    fn=select_image,
+    fn=generate_output,
     inputs=[
-        gr.inputs.Radio(label="Selecciona una imagen:", choices=get_image_choices(), default=get_image_choices()[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")
+        gr.inputs.Textbox(lines=1, label="Fecha de Nacimiento", placeholder="DD/MM/AAAA"),
+        gr.inputs.Image(label="Selecciona una imagen:", type="file")
     ],
     outputs=[
         gr.outputs.Video(label="Respuesta de Andrea (un minuto aproximadamente)").style(width=256),