shaocong commited on
Commit
d39594b
·
2 Parent(s): 837628c e26b013
Files changed (3) hide show
  1. README.md +6 -6
  2. app.py +0 -23
  3. requirements.txt +3 -2
README.md CHANGED
@@ -1,14 +1,14 @@
1
  ---
2
- title: DKT 1
3
- emoji: 🏃
4
- colorFrom: green
5
- colorTo: gray
6
  sdk: gradio
7
- sdk_version: 6.0.2
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
- short_description: DKT-1
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
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
app.py CHANGED
@@ -4,27 +4,6 @@ import os
4
  import os
5
  import subprocess
6
 
7
- def install(package):
8
- subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
9
-
10
- install("torchvision")
11
- install("loguru")
12
- install("imageio")
13
- install("modelscope")
14
- install("einops")
15
- install("safetensors")
16
- install("transformers")
17
- install("ftfy")
18
- install("accelerate")
19
- install("sentencepiece")
20
- install("spaces")
21
- install("opencv-python")
22
- install("trimesh")
23
- install("gradio_litmodel3d")
24
- install("open3d")
25
-
26
-
27
-
28
 
29
 
30
  import gradio as gr
@@ -251,8 +230,6 @@ def load_model_1_3b(device="cuda:0"):
251
 
252
 
253
 
254
-
255
-
256
 
257
 
258
  def get_model(model_size):
 
4
  import os
5
  import subprocess
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
 
9
  import gradio as gr
 
230
 
231
 
232
 
 
 
233
 
234
 
235
  def get_model(model_size):
requirements.txt CHANGED
@@ -9,6 +9,7 @@ modelscope
9
  ftfy
10
  accelerate
11
  loguru
12
- git+https://github.com/microsoft/MoGe.git -i https://pypi.org/simple/ --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host files.pythonhosted.org
13
  sentencepiece
14
- spaces
 
 
 
9
  ftfy
10
  accelerate
11
  loguru
 
12
  sentencepiece
13
+ spaces
14
+
15
+ git+https://github.com/microsoft/MoGe.git -i https://pypi.org/simple/ --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host files.pythonhosted.org