Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,11 @@ from diffusers import StableDiffusion3Pipeline
|
|
3 |
import torch
|
4 |
|
5 |
# 加载模型
|
6 |
-
pipe = StableDiffusion3Pipeline.from_pretrained("
|
7 |
|
8 |
# 定义图像生成函数
|
9 |
def generate_image(prompt):
|
|
|
10 |
return pipe(prompt).images[0]
|
11 |
|
12 |
# 创建 Gradio 界面
|
|
|
3 |
import torch
|
4 |
|
5 |
# 加载模型
|
6 |
+
pipe = StableDiffusion3Pipeline.from_pretrained("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", torch_dtype=torch.float16)
|
7 |
|
8 |
# 定义图像生成函数
|
9 |
def generate_image(prompt):
|
10 |
+
print(prompt)
|
11 |
return pipe(prompt).images[0]
|
12 |
|
13 |
# 创建 Gradio 界面
|