Maj Rag 1

.gitignore

@@ -47,3 +47,5 @@ src/main.py
 src/chat.py
 src/llm/prompt2.py
 src/loaders
+
+###

app.py

@@ -1,3 +1,5 @@
+import spaces
+
 import gradio as gr
 
 from src.vectorstore.faiss_index import load_index
@@ -16,12 +18,16 @@ from pathlib import Path
 
 BASE_DIR = Path(__file__).parent
 
-INDEX_PATH = BASE_DIR / "data/vectorstore/faiss.index"
-METADATA_PATH = BASE_DIR / "data/metadata/metadata.json"
+INDEX_PATH = str(BASE_DIR / "data/vectorstore/faiss.index")
+METADATA_PATH = str(BASE_DIR / "data/metadata/metadata.json")
 
 index = load_index(INDEX_PATH)
 metadata = load_metadata(METADATA_PATH)
 
+@spaces.GPU
+def generate_answer_gpu(prompt):
+    return generate_answer(prompt)
+
 
 def chat_fn(message, history):
 
@@ -49,7 +55,7 @@ Answer:
 """
 
     # 6. Génération réponse LLM
-    answer = generate_answer(prompt)
+    answer = generate_answer_gpu(prompt)
 
     # 6.1. Postprocessing
     answer = clean_answer(answer)

requirements.txt

@@ -9,4 +9,5 @@ pandas==2.3.3
 faiss-cpu==1.14.3
 accelerate==1.14.0
 safetensors==0.8.0
-gradio==6.19.0
+gradio==6.19.0
+spaces==0.50.4