Daniellesry commited on
Commit
aec8db1
·
1 Parent(s): f5052b1
Files changed (2) hide show
  1. README.md +3 -3
  2. app.py +3 -1
README.md CHANGED
@@ -1,14 +1,14 @@
1
  ---
2
- title: DKT 2
3
  emoji: 🖼
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
  sdk_version: 5.44.0
8
  app_file: app.py
9
- pinned: false
10
  license: apache-2.0
11
- short_description: DKT-2
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: DKT
3
  emoji: 🖼
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
  sdk_version: 5.44.0
8
  app_file: app.py
9
+ pinned: true
10
  license: apache-2.0
11
+ short_description: Diffusion Knows Transparency: Repurposing Video Diffusion for Transparent Object Depth and Normal Estimation
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
app.py CHANGED
@@ -287,7 +287,8 @@ def get_model(model_size):
287
 
288
 
289
 
290
- @spaces.GPU(duration=180)
 
291
  @torch.inference_mode()
292
  def process_video(
293
  video_file,
@@ -437,6 +438,7 @@ def process_video(
437
 
438
  # ==== 新增:上下翻转点云 ====
439
  points[:, 2] = -points[:, 2]
 
440
  # =========================
441
 
442
 
 
287
 
288
 
289
 
290
+ # @spaces.GPU(duration=180)
291
+ @spaces.GPU()
292
  @torch.inference_mode()
293
  def process_video(
294
  video_file,
 
438
 
439
  # ==== 新增:上下翻转点云 ====
440
  points[:, 2] = -points[:, 2]
441
+ points[:, 0] = -points[:, 0]
442
  # =========================
443
 
444