Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
import os
|
3 |
import requests
|
4 |
|
5 |
-
API_TOKEN = os.environ.get("HF_HUB_API_TOKEN")
|
6 |
API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
|
7 |
|
8 |
headers = {
|
@@ -27,10 +27,8 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
|
|
27 |
try:
|
28 |
response = requests.post(API_URL, headers=headers, json=payload)
|
29 |
if response.status_code != 200:
|
30 |
-
|
31 |
result = response.json()
|
32 |
-
|
33 |
-
# 判断是否报错
|
34 |
if isinstance(result, dict) and result.get("error"):
|
35 |
return f"[ERROR] {result['error']}"
|
36 |
return result[0]["generated_text"]
|
|
|
2 |
import os
|
3 |
import requests
|
4 |
|
5 |
+
API_TOKEN = os.environ.get("HF_HUB_API_TOKEN") # 或直接写 token
|
6 |
API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
|
7 |
|
8 |
headers = {
|
|
|
27 |
try:
|
28 |
response = requests.post(API_URL, headers=headers, json=payload)
|
29 |
if response.status_code != 200:
|
30 |
+
return f"[HTTP {response.status_code}] {response.text}"
|
31 |
result = response.json()
|
|
|
|
|
32 |
if isinstance(result, dict) and result.get("error"):
|
33 |
return f"[ERROR] {result['error']}"
|
34 |
return result[0]["generated_text"]
|