Update app.py

app.py

@@ -1,38 +1,30 @@
 import gradio as gr
-from transformers import pipeline
-from sympy import sympify, simplify
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-# Modelos de Hugging Face optimizados
-modelos = {
-    "Matemáticas": lambda pregunta: f"Resultado: {simplify(sympify(pregunta))}" if pregunta else "Ingresa una expresión válida.",
-    "Historia": pipeline("text-generation", model="datificate/gpt2-small-spanish"),
-    "Geografía": pipeline("summarization", model="facebook/bart-large-cnn"),
-    "Lengua Española": pipeline("text-generation", model="datificate/gpt2-small-spanish")
-}
+# Modelo Falcon en Hugging Face
+modelo_falcon = "tiiuae/falcon-7b-instruct"  # Puedes cambiarlo por otro más ligero
+tokenizer = AutoTokenizer.from_pretrained(modelo_falcon)
+modelo = AutoModelForCausalLM.from_pretrained(modelo_falcon, device_map="auto")
 
-# Función para procesar la pregunta
-def responder(pregunta, tema):
+# Pipeline de generación de texto optimizado
+generador = pipeline("text-generation", model=modelo, tokenizer=tokenizer, max_new_tokens=100)
+
+# Función de respuesta
+def responder(pregunta):
     if not pregunta.strip():
         return "Por favor, ingresa una pregunta válida."
+    
+    respuesta = generador(pregunta, do_sample=True, max_length=150)
+    return respuesta[0]['generated_text']
 
-    if tema == "Matemáticas":
-        return modelos[tema](pregunta)
-
-    try:
-        respuesta = modelos[tema](pregunta, max_length=50, do_sample=True, return_full_text=True)
-        return respuesta[0]['generated_text']
-    except Exception as e:
-        return f"Error al generar respuesta: {str(e)}"
-
-# Interfaz Gradio mejorada
+# Interfaz en Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("# 📚 Chatbot Multidisciplinario 🧠")
+    gr.Markdown("# 🤖 Chatbot con Falcon")
     
-    tema = gr.Dropdown(choices=list(modelos.keys()), label="Selecciona el Tema", value="Matemáticas")
-    pregunta = gr.Textbox(label="Escribe tu Pregunta", placeholder="Ejemplo: ¿Cuándo fue la Revolución Francesa?")
+    pregunta = gr.Textbox(label="Escribe tu pregunta", placeholder="Ejemplo: ¿Qué es la inteligencia artificial?")
     respuesta = gr.Textbox(label="Respuesta", interactive=False)
     
     boton = gr.Button("Enviar")
-    boton.click(responder, inputs=[pregunta, tema], outputs=respuesta)
+    boton.click(responder, inputs=pregunta, outputs=respuesta)
 
 demo.launch()