Update app.py

app.py

@@ -4,15 +4,18 @@ from datasets import load_dataset
 # Carga el modelo
 model = gr.load("models/Qwen/Qwen2.5-Coder-32B-Instruct")
 
-# Carga el dataset
-dataset = load_dataset("Eim/laravel-docs")
+# Carga el split de entrenamiento del dataset
+train_dataset = load_dataset("yannelli/laravel-11-qa", split="train")
+
+# Inspección opcional para verificar la estructura
+print(train_dataset[0])
 
 # Función para procesar las entradas del chatbot
 def chatbot(input_text):
-    # Filtra el dataset para obtener información relevante
+    # Reemplaza "content" con la clave correcta según la inspección
     relevant_docs = [
-        doc["content"] for doc in dataset["train"] 
-        if input_text.lower() in doc["content"].lower()
+        doc["text"] for doc in train_dataset
+        if input_text.lower() in doc["text"].lower()
     ]
     
     # Respuesta del modelo