Update app.py

app.py

@@ -1,7 +1,4 @@
-# Primero, instalamos las bibliotecas necesarias
-# NOTA: No uses el signo "!" en Hugging Face, ya que eso genera un error.
-# Si usas Hugging Face Spaces, las dependencias se instalan desde el archivo requirements.txt.
-# Asegúrate de tener este archivo con las líneas:
+# Instalar bibliotecas en requirements.txt (sin usar `!` en el código de app.py):
 # transformers
 # gradio
 # torch
@@ -9,34 +6,28 @@
 from transformers import pipeline
 import gradio as gr
 
-# Configuración del modelo de lenguaje: distilgpt2 para respuestas rápidas y concisas
-chatbot = pipeline(
-    "text-generation",
-    model="distilgpt2",    # Usa "distilgpt2" para respuestas rápidas
-    max_length=50,
-    pad_token_id=50256,
-    temperature=0.5,       # Baja aleatoriedad para respuestas más coherentes
-    top_k=50
-)
+# Configuración del modelo de respuesta a preguntas
+qa_pipeline = pipeline("question-answering", model="deepset/bert-base-cased-squad2")
 
-# Definimos la función para manejar la pregunta y limpiar la respuesta
+# Función para procesar la pregunta
 def responder(pregunta):
-    # Generamos la respuesta con el modelo
-    respuesta = chatbot(pregunta, max_length=50, num_return_sequences=1)
-    respuesta_texto = respuesta[0]['generated_text']
-    
-    # Limpiamos la respuesta para eliminar repeticiones
-    respuesta_texto = respuesta_texto.split(".")[0] + "."
-    
-    return respuesta_texto
+    # Definir el contexto en el que el modelo busca la respuesta
+    contexto = """
+    Francia es un país cuyo capital es París. Fue fundada hace varios siglos y es famosa por su cultura,
+    arte, gastronomía y historia. París, la capital, es conocida por monumentos como la Torre Eiffel,
+    el Museo del Louvre, y por ser una de las ciudades más visitadas del mundo.
+    """
+
+    respuesta = qa_pipeline(question=pregunta, context=contexto)
+    return respuesta['answer']
 
 # Interfaz gráfica de usuario usando Gradio
 iface = gr.Interface(
     fn=responder,
     inputs="text",
     outputs="text",
-    title="Chatbot AI",
-    description="Hazme cualquier pregunta, y te responderé como un asistente de IA."
+    title="Chatbot de Respuestas Rápidas",
+    description="Pregunta lo que quieras, y te responderé con información específica."
 )
 
 # Ejecutar la aplicación