Update app.py

app.py

@@ -10,7 +10,7 @@ generator = pipeline('text-generation', model="checkpoints")
 tokenizer = GPT2Tokenizer.from_pretrained('checkpoints')
 os.environ["TOKENIZERS_PARALLELISM"] = "true"
 
-def generate_output(name, date_of_birth):
+def generate_output(name, date_of_birth, progress=gr.Progress()):
     prompt = f"Bienvenido {name}:"
     input_tokens = tokenizer.encode(prompt, add_special_tokens=True)
     input_text = tokenizer.decode(input_tokens)
@@ -20,6 +20,10 @@ def generate_output(name, date_of_birth):
     if len(gpt2_output) == 0 or 'generated_text' not in gpt2_output[0]:
         return None, "No se pudo generar el texto."
 
+    def _progress(generated, to_generate):
+        progress((generated, to_generate))
+    _progress(0, 1)
+
     try:
         tts = gTTS(generated_text, lang='es')
         temp_audio_path = "temp_audio.mp3"
@@ -28,6 +32,7 @@ def generate_output(name, date_of_birth):
         audio = AudioSegment.from_mp3(temp_audio_path)
         audio.export(audio_path, format="wav")
         print("Archivo de audio generado:", audio_path)
+        _progress(1, 2)
     except Exception as e:
         return None, f"No se pudo generar el audio: {str(e)}"
 
@@ -41,6 +46,7 @@ def generate_output(name, date_of_birth):
     os.remove(temp_audio_path)
 
     if os.path.isfile(output_video_path):
+        _progress(2, 2)
         return output_video_path, None
     return None, "No se pudo generar el video"