yaoqifa commited on
Commit
c10de4e
·
verified ·
1 Parent(s): cda016d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,6 +41,7 @@ RESOLUTION_MAP = {
41
  }
42
 
43
  IP_REQUEST_HISTORY = {}
 
44
 
45
  # ==========================================
46
  # 2. 辅助功能
@@ -190,7 +191,7 @@ async def core_generate(req: GenerateRequest):
190
  final_prompt = f"{req.prompt} {suffix}"
191
 
192
  # 3. API 请求
193
- client = AsyncOpenAI(api_key=req.api_key, base_url="https://api.cloudapp.ink/v1", max_retries=0, timeout=60.0)
194
  img_b = None
195
 
196
  if "gemini" in req.model.lower():
 
41
  }
42
 
43
  IP_REQUEST_HISTORY = {}
44
+ app_base_url = os.getenv("APP_BASE_URL", "").strip()
45
 
46
  # ==========================================
47
  # 2. 辅助功能
 
191
  final_prompt = f"{req.prompt} {suffix}"
192
 
193
  # 3. API 请求
194
+ client = AsyncOpenAI(api_key=req.api_key, base_url=f"{app_base_url}/v1", max_retries=0, timeout=360.0)
195
  img_b = None
196
 
197
  if "gemini" in req.model.lower():