Create app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from diffusers import DiffusionPipeline
+
+# 加载模型
+pipeline = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion-fp8")
+
+# 定义生成函数
+def generate_image(prompt):
+    image = pipeline(prompt).images[0]
+    return image
+
+# 创建 Gradio 界面
+interface = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="输入提示词"),
+    outputs=gr.Image(label="生成的图像"),
+    title="Shuttle-3 Diffusion FP8 图像生成",
+    description="输入提示词，生成图像。"
+)
+
+# 启动应用
+interface.launch()