Spaces:
Running
on
A10G
Running
on
A10G
Update
Browse files- Dockerfile +1 -1
- 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.
|
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
|
|
|
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):
|