gengs commited on
Commit
b8c5f04
1 Parent(s): 53b7789

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,10 +3,11 @@ from diffusers import StableDiffusion3Pipeline
3
  import torch
4
 
5
  # 加载模型
6
- pipe = StableDiffusion3Pipeline.from_pretrained("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA", torch_dtype=torch.float16)
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 界面