feat: add model

app.py

@@ -6,11 +6,21 @@
 @Project : Faice_text2face
 """
 import gradio as gr
+from stable_diffusion_inference import inference
 
 
-def greet(name):
-    return "Hello " + name + "!!"
+# def greet(name):
+#     return "Hello " + name + "!!"
+#
+#
+# demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def text2face(prompt):
+    image = inference(prompt)
+    return image
+
+
+demo = gr.Interface(fn=text2face, inputs="text", outputs="image")
+
 demo.launch()

stable_diffusion_inference.py

@@ -0,0 +1,19 @@
+# -*- coding: UTF-8 -*-
+"""
+@Time : 28/05/2025 11:56
+@Author : xiaoguangliang
+@File : stable_diffusion_inference.py
+@Project : Faice_text2face
+"""
+import torch
+from diffusers import StableDiffusionPipeline
+
+model_path = 'Ngene787/Faice_text2face'
+
+
+def inference(prompt):
+    pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+    # pipe = pipe.to("cuda")
+
+    image = pipe(prompt).images[0]
+    return image