58 lines
694 B
Python
58 lines
694 B
Python
import base64
|
|
|
|
import requests
|
|
|
|
invoke_url = "http://localhost:8003/v1/infer"
|
|
|
|
|
|
|
|
payload = {
|
|
|
|
"text_prompts": [
|
|
|
|
{
|
|
|
|
"text": "realistic futuristic city-downtown with short buildings, sunset",
|
|
|
|
"weight": 1
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"text": "" ,
|
|
|
|
"weight": -1
|
|
|
|
}
|
|
|
|
],
|
|
|
|
"cfg_scale": 5,
|
|
|
|
"sampler": "K_DPM_2_ANCESTRAL",
|
|
|
|
"seed": 0,
|
|
|
|
"steps": 25
|
|
|
|
}
|
|
|
|
|
|
|
|
response = requests.post(invoke_url, json=payload)
|
|
|
|
response.raise_for_status()
|
|
|
|
data = response.json()
|
|
|
|
img_base64 = data['artifacts'][0]["base64"]
|
|
|
|
img_bytes = base64.b64decode(img_base64)
|
|
|
|
|
|
|
|
with open("test_image.jpg", "wb") as f:
|
|
f.write(img_bytes)
|
|
|
|
|