hysts HF staff commited on
Commit
7d275a2
1 Parent(s): dc0c2a7
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. model.py +3 -2
Dockerfile CHANGED
@@ -45,7 +45,7 @@ RUN pip install --no-cache-dir -U torch==1.12.1+cu113 torchvision==0.13.1+cu113
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
  RUN pip install --no-cache-dir -U kaolin==0.13.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-1.12.1_cu113.html
48
- RUN pip install --no-cache-dir -U gradio==3.17.1
49
 
50
  COPY --chown=1000 . ${HOME}/app
51
  RUN cd TEXTurePaper && patch -p1 < ../patch
 
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
  RUN pip install --no-cache-dir -U kaolin==0.13.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-1.12.1_cu113.html
48
+ RUN pip install --no-cache-dir -U gradio==3.30.0
49
 
50
  COPY --chown=1000 . ${HOME}/app
51
  RUN cd TEXTurePaper && patch -p1 < ../patch
model.py CHANGED
@@ -5,7 +5,7 @@ import pathlib
5
  import shlex
6
  import subprocess
7
  import sys
8
- from typing import Generator
9
 
10
  import gradio as gr
11
 
@@ -51,7 +51,8 @@ class Model:
51
 
52
  def run(
53
  self, shape_path: str, text: str, seed: int, guidance_scale: float
54
- ) -> Generator[tuple[list[str], str | None, str | None, str], None, None]:
 
55
  if not shape_path.endswith('.obj'):
56
  raise gr.Error('The input file is not .obj file.')
57
  if not self.check_num_faces(shape_path):
 
5
  import shlex
6
  import subprocess
7
  import sys
8
+ from typing import Generator, Optional
9
 
10
  import gradio as gr
11
 
 
51
 
52
  def run(
53
  self, shape_path: str, text: str, seed: int, guidance_scale: float
54
+ ) -> Generator[tuple[list[str], Optional[str], Optional[str], str], None,
55
+ None]:
56
  if not shape_path.endswith('.obj'):
57
  raise gr.Error('The input file is not .obj file.')
58
  if not self.check_num_faces(shape_path):