Update app.py

app.py

@@ -48,13 +48,13 @@ def store_special_tokens(tokenizer, model_name):
 def load_special_tokens(tokenizer, model_name):
     special_tokens = redis_client.hgetall(f"tokenizer_special_tokens:{model_name}")
     if special_tokens:
-        tokenizer.pad_token = special_tokens.get('pad_token')
+        tokenizer.pad_token = special_tokens.get('pad_token', '').decode("utf-8")
         tokenizer.pad_token_id = int(special_tokens.get('pad_token_id', -1))
-        tokenizer.eos_token = special_tokens.get('eos_token')
+        tokenizer.eos_token = special_tokens.get('eos_token', '').decode("utf-8")
         tokenizer.eos_token_id = int(special_tokens.get('eos_token_id', -1))
-        tokenizer.unk_token = special_tokens.get('unk_token')
+        tokenizer.unk_token = special_tokens.get('unk_token', '').decode("utf-8")
         tokenizer.unk_token_id = int(special_tokens.get('unk_token_id', -1))
-        tokenizer.bos_token = special_tokens.get('bos_token')
+        tokenizer.bos_token = special_tokens.get('bos_token', '').decode("utf-8")
         tokenizer.bos_token_id = int(special_tokens.get('bos_token_id', -1))
 
 def train_and_store_transformers_model(model_name, data):
@@ -299,7 +299,7 @@ def generate_response_from_prompt(prompt, model_name="google/flan-t5-xl"):
     responses = generate_random_response([prompt], generator)
     return responses[0]
 
-def generate_image_from_prompt(prompt, image_type, model_name):
+def generate_image_from_prompt(prompt, image_type, model_name="CompVis/stable-diffusion-v1-4"):
     if image_type == "diffusers":
         image = generate_diffusers_image_from_redis(model_name, prompt)
     elif image_type == "stable-diffusion":
@@ -321,9 +321,10 @@ def gradio_app():
         with gr.Row():
             prompt_image = gr.Textbox(label="Prompt de Imagen")
             image_type = gr.Dropdown(["diffusers", "stable-diffusion", "img2img"], label="Tipo de Imagen")
+            model_name_image = gr.Textbox(label="Nombre del Modelo", value="CompVis/stable-diffusion-v1-4")
             image_output = gr.Image(label="Imagen Generada")
             image_button = gr.Button("Generar Imagen")
-        image_button.click(generate_image_from_prompt, inputs=[prompt_image, image_type], outputs=image_output)
+        image_button.click(generate_image_from_prompt, inputs=[prompt_image, image_type, model_name_image], outputs=image_output)
 
         gr.Markdown("## Generación de Video")
         with gr.Row():
@@ -334,10 +335,11 @@ def gradio_app():
 
         gr.Markdown("## Generación de Audio con MusicGen")
         with gr.Row():
+            model_name_audio = gr.Textbox(label="Nombre del Modelo", value="facebook/musicgen-small")
             text_prompts_audio = gr.Textbox(label="Prompts de Audio")
             audio_output = gr.Audio(label="Audio Generado")
             audio_button = gr.Button("Generar Audio")
-        audio_button.click(generate_musicgen_audio_from_redis, inputs=text_prompts_audio, outputs=audio_output)
+        audio_button.click(generate_musicgen_audio_from_redis, inputs=[model_name_audio, text_prompts_audio], outputs=audio_output)
 
         gr.Markdown("## Transcripción de Audio con Whisper")
         with gr.Row():
@@ -348,19 +350,21 @@ def gradio_app():
 
         gr.Markdown("## Traducción de Texto")
         with gr.Row():
+            model_name_translate = gr.Textbox(label="Nombre del Modelo", value="Helsinki-NLP/opus-mt-en-es")
             text_input = gr.Textbox(label="Texto a Traducir")
             translation_output = gr.Textbox(label="Traducción")
             src_lang_input = gr.Textbox(label="Idioma de Origen", value="en")
             tgt_lang_input = gr.Textbox(label="Idioma de Destino", value="es")
             translate_button = gr.Button("Traducir Texto")
-        translate_button.click(translate_text_from_redis, inputs=[text_input, src_lang_input, tgt_lang_input], outputs=translation_output)
+        translate_button.click(translate_text_from_redis, inputs=[model_name_translate, text_input, src_lang_input, tgt_lang_input], outputs=translation_output)
 
         gr.Markdown("## Resumen de Texto")
         with gr.Row():
+            model_name_summarize = gr.Textbox(label="Nombre del Modelo", value="facebook/bart-large-cnn")
             text_to_summarize = gr.Textbox(label="Texto para Resumir")
             summary_output = gr.Textbox(label="Resumen")
             summarize_button = gr.Button("Generar Resumen")
-        summarize_button.click(summarize_text_from_redis, inputs=text_to_summarize, outputs=summary_output)
+        summarize_button.click(summarize_text_from_redis, inputs=[model_name_summarize, text_to_summarize], outputs=summary_output)
 
     app.launch()