smgc commited on
Commit
5aaa8d7
·
verified ·
1 Parent(s): e91904d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -182,27 +182,29 @@ def handle_request():
182
  else:
183
  enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
184
 
185
- new_url = f'https://api.siliconflow.cn/v1/{mapped_model}/text-to-image'
186
  new_request_body = {
 
187
  "prompt": enhanced_prompt,
 
188
  "image_size": image_size,
189
  "batch_size": 1,
 
190
  "num_inference_steps": 4,
191
  "guidance_scale": 1
192
  }
193
 
194
  headers = {
195
- 'accept': 'application/json',
196
- 'content-type': 'application/json',
197
- 'Authorization': f'Bearer {random_token}'
198
  }
199
-
200
  response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
201
  response.raise_for_status()
202
  response_body = response.json()
203
-
204
- if 'images' in response_body and response_body['images'] and 'url' in response_body['images'][0]:
205
- image_url = response_body['images'][0]['url']
206
  else:
207
  raise ValueError("Unexpected response structure from image generation API")
208
 
 
182
  else:
183
  enhanced_prompt = translate_and_enhance_prompt(clean_prompt, random_token)
184
 
185
+ new_url = 'https://api.siliconflow.cn/v1/images/generations'
186
  new_request_body = {
187
+ "model": mapped_model, # 使用映射后的模型名称
188
  "prompt": enhanced_prompt,
189
+ "negative_prompt": "", # 如果不需要负提示词,可以留空
190
  "image_size": image_size,
191
  "batch_size": 1,
192
+ "seed": random.randint(0, 4999999999), # 随机生成一个种子值
193
  "num_inference_steps": 4,
194
  "guidance_scale": 1
195
  }
196
 
197
  headers = {
198
+ 'Authorization': f'Bearer {random_token}',
199
+ 'Content-Type': 'application/json'
 
200
  }
201
+
202
  response = requests.post(new_url, headers=headers, json=new_request_body, timeout=60)
203
  response.raise_for_status()
204
  response_body = response.json()
205
+
206
+ if 'data' in response_body and response_body['data'] and 'url' in response_body['data'][0]:
207
+ image_url = response_body['data'][0]['url']
208
  else:
209
  raise ValueError("Unexpected response structure from image generation API")
210