Upload app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-# Cargar el modelo de traducción
-translator = pipeline("translation", model="projecte-aina/aina-translator-es-ast")
+# Cargar el tokenizador y el modelo correctamente
+tokenizer = AutoTokenizer.from_pretrained("projecte-aina/aina-translator-es-ast")
+model = AutoModelForSeq2SeqLM.from_pretrained("projecte-aina/aina-translator-es-ast")
+
+# Crear el pipeline de traducción
+translator = pipeline("translation", model=model, tokenizer=tokenizer)
 
 # Función de chatbot que traduce automáticamente
 def chatbot_response(message, chat_history):
@@ -18,7 +22,7 @@ with gr.Blocks() as demo:
     msg = gr.Textbox(label="Escribe tu mensaje en Español...")
     send_btn = gr.Button("Enviar")
 
-    # Asegurar que la función está bien escrita
+    # Configuración del botón de envío
     send_btn.click(fn=chatbot_response, inputs=[msg, chatbot], outputs=[msg, chatbot])
 
 # Lanzar la aplicación