asahi417 commited on
Commit
abd7e3e
·
verified ·
1 Parent(s): 92d00b1

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
- import torch
3
  from diffusers.utils import load_image
4
  import spaces
5
  from panna.pipeline import PipelineDepth2ImageV2
@@ -13,7 +12,7 @@ for n in range(1, 10):
13
  example_files.append(f"demo{n:0>2}.jpg")
14
 
15
 
16
- @spaces.GPU
17
  def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
18
  return model(
19
  init_image,
 
1
  import gradio as gr
 
2
  from diffusers.utils import load_image
3
  import spaces
4
  from panna.pipeline import PipelineDepth2ImageV2
 
12
  example_files.append(f"demo{n:0>2}.jpg")
13
 
14
 
15
+ @spaces.GPU(duration=120)
16
  def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
17
  return model(
18
  init_image,