Update app.py

app.py

@@ -1,6 +1,26 @@
 import gradio as gr
+from diffusers import StableDiffusion3Pipeline
+import torch
+
+# 加载模型
+pipe = StableDiffusion3Pipeline.from_pretrained("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA", torch_dtype=torch.float16)
+
+# 定义图像生成函数
+def generate_image(prompt):
+    return pipe(prompt).images[0]
+
+# 创建 Gradio 界面
+iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
+
+# 启动界面
+iface.launch()
+
+
+# import gradio as gr
+
+# gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA").launch()
+
 
-gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA").launch()
 
 # import gradio as gr
 # import torch