change app.py

app.py

@@ -1,7 +1,27 @@
+# app.py
 import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-def greet(name):
-    return "Hello " + name + "!!"
+# 初始化模型（免费版Space选择小模型）
+model = StableDiffusionPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4",
+    torch_dtype=torch.float16,
+    use_safetensors=True,
+    low_cpu_mem_usage=True  # 节省内存
+).to("cuda") if torch.cuda.is_available() else None  # 兼容CPU模式
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate(prompt):
+    if model is None:
+        return "当前环境不支持图像生成"
+    image = model(prompt, num_inference_steps=15).images[0]
+    return image
+
+# 基础界面
+interface = gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(label="输入表情包描述"),
+    outputs=gr.Image(label="生成结果"),
+    examples=[["一只生气的猫"], ["跳舞的香蕉"]]
+)
+interface.launch()