Spaces:
Sleeping
Sleeping
Commit
·
e251c37
1
Parent(s):
956a22b
debug
Browse files
app.py
CHANGED
@@ -3,6 +3,16 @@
|
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
from app_canny import create_demo as create_demo_canny
|
7 |
from app_texnet import create_demo as create_demo_texnet
|
8 |
|
@@ -17,15 +27,6 @@ if not torch.cuda.is_available():
|
|
17 |
# model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="Canny")
|
18 |
model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="texnet")
|
19 |
|
20 |
-
import sys
|
21 |
-
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
22 |
-
version_str="".join([
|
23 |
-
f"py3{sys.version_info.minor}_cu",
|
24 |
-
torch.version.cuda.replace(".",""),
|
25 |
-
f"_pyt{pyt_version_str}"
|
26 |
-
])
|
27 |
-
print(f"Using version: {version_str}")
|
28 |
-
|
29 |
with gr.Blocks() as demo:
|
30 |
gr.Markdown(DESCRIPTION)
|
31 |
gr.DuplicateButton(
|
|
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
|
6 |
+
import sys
|
7 |
+
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
8 |
+
version_str="".join([
|
9 |
+
f"py3{sys.version_info.minor}_cu",
|
10 |
+
torch.version.cuda.replace(".",""),
|
11 |
+
f"_pyt{pyt_version_str}"
|
12 |
+
])
|
13 |
+
print(f"Using version: {version_str}") # used to locate pytorch3d version in the requirements.txt for huggingface
|
14 |
+
|
15 |
+
|
16 |
from app_canny import create_demo as create_demo_canny
|
17 |
from app_texnet import create_demo as create_demo_texnet
|
18 |
|
|
|
27 |
# model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="Canny")
|
28 |
model = Model(base_model_id=DEFAULT_MODEL_ID, task_name="texnet")
|
29 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
with gr.Blocks() as demo:
|
31 |
gr.Markdown(DESCRIPTION)
|
32 |
gr.DuplicateButton(
|