Update app.py

app.py

@@ -5,18 +5,19 @@ from transformers import AutoTokenizer, AutoModel, pipeline
 import torch
 import faiss
 import numpy as np
-from spaces import GPU  # IMPORTANTE para ZeroGPU
+from spaces import GPU  # Importante para ZeroGPU
 
-# Token para modelos privados si se requiere
+# Token para modelos privados
 hf_token = os.getenv("HF_KEY")
 
 # Cargar índice FAISS y los chunks
 if os.path.exists("index.pkl"):
-    # justo después de cargar el index y los chunks
     with open("index.pkl", "rb") as f:
-    index, chunks = pickle.load(f)
+        index, chunks = pickle.load(f)
+else:
+    raise FileNotFoundError("No se encontró el archivo 'index.pkl'.")
 
-# Filtro suave de limpieza (nuevo punto C)
+# Filtro suave de limpieza (punto C)
 def limpiar_chunks(chunks):
     palabras_legales = ["artículo", "sanción", "pena", "multa", "prohibido", "infracción", "autoridad", "norma"]
     nuevos_chunks = []
@@ -52,7 +53,7 @@ llm = pipeline(
     trust_remote_code=True
 )
 
-@GPU  # DECORADOR para que ZeroGPU ejecute esta función en CPU/GPU remota
+@GPU  # Activar ZeroGPU
 def responder(pregunta):
     if not pregunta:
         return "Por favor ingresa una pregunta."
@@ -111,4 +112,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()