feat: add inference_api

app.py

@@ -7,7 +7,8 @@
 """
 import gradio as gr
 
-from stable_diffusion_inference import inference, MAX_SEED
+from stable_diffusion_inference import MAX_SEED
+from inference_api import inference
 from utils import timer
 
 MAX_IMAGE_SIZE = 1024

inference_api.py

@@ -0,0 +1,28 @@
+# -*- coding: UTF-8 -*-
+"""
+@Time : 30/05/2025 10:45
+@Author : xiaoguangliang
+@File : inference_api.py
+@Project : Faice_text2face
+"""
+from gradio_client import Client
+
+client = Client("https://e517e907a9e4213655.gradio.live/")
+
+
+def inference(prompt,
+              negative_prompt="",
+              seed=0,
+              randomize_seed=False,
+              guidance_scale=7.5,
+              num_inference_steps=20):
+    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"
+    )
+    return result