jingyangcarl commited on
Commit
e251c37
·
1 Parent(s): 956a22b
Files changed (1) hide show
  1. app.py +10 -9
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(