Hugging Face's logo

Spaces:
CHUNYU0505
/
RAG_Test_System
Sleeping

App Files Community
CHUNYU0505 commited on
Commit
9c1b3ba
·
verified ·
1 Parent(s): c4310e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -25,9 +25,11 @@ else:
25
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 5})
26
 
27
  # -------------------------------
28
- # 1. 中文模型（Randeng-T5）
29
  # -------------------------------
30
- MODEL_NAME = "IDEA-CCNL/Randeng-T5-784M-Summary-Chinese"
 
 
31
 
32
  HF_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
33
  if HF_TOKEN:
 
25
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 5})
26
 
27
  # -------------------------------
28
+ # 1. 中文模型（T5 Pegasus）
29
  # -------------------------------
30
+ MODEL_NAME = "imxly/t5-pegasus-small"
31
+
32
+
33
 
34
  HF_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
35
  if HF_TOKEN: