Update app.py

app.py

@@ -4,11 +4,10 @@ from transformers import pipeline
 # Cargar el modelo de DeepSeek AI para el chatbot
 @st.cache_resource
 def load_model():
-    # Asegúrate de que el nombre del modelo es correcto o ajusta según el que uses
     model_name = "deepseek-ai/DeepSeek-R1"  # Cambia esto si el modelo es diferente
     
-    # Agregar trust_remote_code=True para permitir la ejecución del código remoto
-    chatbot = pipeline("conversational", model=model_name, trust_remote_code=True)
+    # Usar la tarea 'text-generation' en lugar de 'conversational'
+    chatbot = pipeline("text-generation", model=model_name, trust_remote_code=True)
     return chatbot
 
 # Inicializa el modelo
@@ -26,7 +25,7 @@ def main():
 
     if user_input:
         # Generar una respuesta del chatbot usando el modelo
-        response = chatbot(user_input)
+        response = chatbot(user_input, max_length=100, num_return_sequences=1)
         
         # Mostrar la respuesta generada
         st.write(f"**Respuesta del chatbot:** {response[0]['generated_text']}")