Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
 import numpy as np
-import os
 from sentence_transformers import SentenceTransformer
 from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
 import os
+from huggingface_hub import hf_hub_download
+import faiss
 
 # 设置内存交换参数
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32"
@@ -62,6 +63,23 @@ with gr.Blocks() as demo:
         outputs=output
     )
 
+# 在应用启动时下载索引
+INDEX_PATH = hf_hub_download(
+    repo_id="GOGO198/GOGO_rag_index",
+    filename="faiss_index.bin",
+    cache_dir="/data"
+)
+
+METADATA_PATH = hf_hub_download(
+    repo_id="GOGO198/GOGO_rag_index",
+    filename="metadata.csv",
+    cache_dir="/data"
+)
+
+# 加载索引
+index = faiss.read_index(INDEX_PATH)
+metadata = pd.read_csv(METADATA_PATH)
+    
 # 启动应用
 demo.launch(
     server_name="0.0.0.0",