zhufengyun commited on
Commit
13c9d42
·
verified ·
1 Parent(s): 211d822

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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
- return f"[HTTP {response.status_code}] {response.text}" # 提示HTTP错误或报错内容
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"]