Spaces:
Sleeping
Sleeping
File size: 1,522 Bytes
fa8c1f4 48a2cde 0c4e4fb fa8c1f4 4d24ceb 0c4e4fb f2aabd5 fa8c1f4 4d24ceb fa8c1f4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# -*- coding: UTF-8 -*-
"""
@Time : 30/05/2025 10:45
@Author : xiaoguangliang
@File : inference_api.py
@Project : Faice_text2face
"""
from gradio_client import Client
inference_url = "https://29f185caa56dd00128.gradio.live"
def api_unconditional(seed,
randomize_seed=False,
num_inference_steps=20):
client = Client(inference_url)
result = client.predict(
seed=seed,
randomize_seed=randomize_seed,
num_inference_steps=num_inference_steps,
api_name="/inference_unconditional"
)
return result
def api_class_guidance(label_name,
seed=0,
randomize_seed=False,
num_inference_steps=20):
client = Client(inference_url)
result = client.predict(
label_name=label_name,
seed=seed,
randomize_seed=randomize_seed,
num_inference_steps=num_inference_steps,
api_name="/inference_class_guidance"
)
return result
def api_sd(prompt,
negative_prompt="",
seed=0,
randomize_seed=False,
guidance_scale=7.5,
num_inference_steps=20):
client = Client(inference_url)
result = client.predict(
prompt=prompt,
negative_prompt=negative_prompt,
seed=seed,
randomize_seed=randomize_seed,
guidance_scale=guidance_scale,
num_inference_steps=num_inference_steps,
api_name="/inference_sd"
)
return result
|