hanshu.yan commited on
Commit
e4ec328
β€’
1 Parent(s): 0fddefd

update dep

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,9 +1,12 @@
1
- import os, logging, time, argparse, random, tempfile, rembg
2
  import gradio as gr
3
  import numpy as np
4
  import torch
5
  from PIL import Image
6
  from functools import partial
 
 
 
7
  from tsr.system import TSR
8
  from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
9
 
 
1
+ import os, logging, time, argparse, random, tempfile, rembg, shlex, subprocess
2
  import gradio as gr
3
  import numpy as np
4
  import torch
5
  from PIL import Image
6
  from functools import partial
7
+
8
+ subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))
9
+
10
  from tsr.system import TSR
11
  from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
12