Update app.py

app.py

@@ -2,7 +2,6 @@ from flask import Flask, request, jsonify, Response
 import requests
 import json
 import time
-import random
 
 app = Flask(__name__)
 
@@ -32,24 +31,10 @@ def handle_request():
             "guidance_scale": 1
         }
 
-        # 从传入的 Authorization 头中随机选择一个 token
-        authorization_header = request.headers.get('Authorization')
-        if authorization_header:
-            # 去掉 "Bearer " 前缀并分割 token
-            tokens = authorization_header.replace("Bearer ", "").split(',')
-            if len(tokens) > 1:
-                selected_token = random.choice(tokens).strip()
-            else:
-                selected_token = tokens[0].strip()
-            # 重新格式化为 "Bearer 随机选择的token"
-            selected_token = f"Bearer {selected_token}"
-        else:
-            return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401
-
         headers = {
             'accept': 'application/json',
             'content-type': 'application/json',
-            'Authorization': selected_token
+            'Authorization': request.headers.get('Authorization')
         }
 
         response = requests.post(new_url, headers=headers, json=new_request_body)