Spaces:
Sleeping
Sleeping
改用有 Inference API 的模型
Browse files
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-
|
| 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 |
|