ethe commited on
Commit
24d448e
·
1 Parent(s): 84b2c1e

modify local file only and add inputs img box

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ pipe = StableDiffusionPipeline.from_single_file(
7
  "https://huggingface.co/ethe/Architecture_model/blob/main/architectureExterior_v40Exterior.safetensors",
8
  # "/mnt/pfs-guan-ssai/cv/panxuhao/checkpoints/stable-diffusion-xl-base-1.0/sd_xl_base_1.0.safetensors",
9
  torch_dtype=torch.float16,
10
- local_files_only=True,
11
  variant="fp16",
12
  use_safetensors=True,
13
  )
@@ -31,7 +31,7 @@ def inference():
31
  return image
32
  demo = gr.Interface(
33
  fn=inference,
34
- inputs=None,
35
  outputs=gr.Gallery(label="可能满足你需求的室内设计图:",
36
  columns=3,
37
  height="auto",
 
7
  "https://huggingface.co/ethe/Architecture_model/blob/main/architectureExterior_v40Exterior.safetensors",
8
  # "/mnt/pfs-guan-ssai/cv/panxuhao/checkpoints/stable-diffusion-xl-base-1.0/sd_xl_base_1.0.safetensors",
9
  torch_dtype=torch.float16,
10
+ # local_files_only=True,
11
  variant="fp16",
12
  use_safetensors=True,
13
  )
 
31
  return image
32
  demo = gr.Interface(
33
  fn=inference,
34
+ inputs=gr.Image(height=512, width=512, label="参考图片"),
35
  outputs=gr.Gallery(label="可能满足你需求的室内设计图:",
36
  columns=3,
37
  height="auto",