CHUNYU0505 commited on
Commit
9044d56
·
verified ·
1 Parent(s): 4d5736b

改用有 Inference API 的模型

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 5})
52
  # -------------------------------
53
  # 4. 定義 REST API 呼叫函數
54
  # -------------------------------
55
- INFERENCE_MODEL = "google/flan-t5-large"
56
  API_URL = f"https://api-inference.huggingface.co/models/{INFERENCE_MODEL}"
57
  HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
58
 
 
52
  # -------------------------------
53
  # 4. 定義 REST API 呼叫函數
54
  # -------------------------------
55
+ INFERENCE_MODEL = "google/flan-t5-xl"
56
  API_URL = f"https://api-inference.huggingface.co/models/{INFERENCE_MODEL}"
57
  HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
58