Update app.py

app.py

@@ -8,7 +8,11 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
 
 # Cargar tu conjunto de datos
-dataset = load_dataset('csv', data_files='alpaca.csv')
+try:
+    dataset = load_dataset('csv', data_files='alpaca.csv')
+    print("Conjunto de datos cargado correctamente.")
+except Exception as e:
+    print(f"Error al cargar el conjunto de datos: {e}")
 
 # Preprocesar los datos
 def preprocess_function(examples):
@@ -58,11 +62,21 @@ chat_history_ids = None
 # Función de chat
 def chat_with_bot(user_input):
     global chat_history_ids
-    new_user_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
-    bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if chat_history_ids is not None else new_user_input_ids
-    chat_history_ids = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
-    return tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
+    try:
+        new_user_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
+        bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if chat_history_ids is not None else new_user_input_ids
+        chat_history_ids = model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+        response = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
+        
+        # Si la respuesta es vacía o no tiene sentido, devuelve una respuesta predeterminada
+        if not response.strip():
+            return "Lo siento, no entiendo la pregunta."
+        
+        return response
+    except Exception as e:
+        return f"Error: {e}. No pude procesar tu pregunta."
 
 # Crear la interfaz de Gradio
 iface = gr.Interface(fn=chat_with_bot, inputs="text", outputs="text", title="Chatbot Entrenado")
 iface.launch()
+