Spaces:
Build error
Build error
| # -*- 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 | |