Update app.py

app.py

@@ -5,7 +5,6 @@ from gtts import gTTS
 from pydub import AudioSegment
 from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
 from PIL import Image
-import random
 
 generator = pipeline('text-generation', model="checkpoints")
 tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
@@ -32,8 +31,7 @@ def generate_output(name, date_of_birth):
     except Exception as e:
         return None, f"No se pudo generar el audio: {str(e)}"
 
-    random_image = select_random_image()
-    command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face {random_image} --audio audio.wav --outfile video.mp4 --nosmooth"
+    command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face oraculo.jpg --audio audio.wav --outfile video.mp4 --nosmooth"
     process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if process.returncode != 0:
         error_message = process.stderr.decode("utf-8")
@@ -46,15 +44,6 @@ def generate_output(name, date_of_birth):
         return output_video_path, None
     return None, "No se pudo generar el video"
 
-def select_random_image():
-    return random.choice([f"{i}.jpg" for i in range(1, 3)])
-
-def error_message_fn(error_message):
-    if error_message is not None:
-        return gr.outputs.Textbox(text=error_message, placeholder="Error")
-    else:
-        return None
-
 iface = gr.Interface(
     fn=generate_output,
     inputs=[
@@ -69,4 +58,4 @@ iface = gr.Interface(
     description="Por favor, ingresa tu nombre y fecha de nacimiento."
 )
 
-iface.launch()
+iface.launch()