xiexh20 commited on
Commit
1963f53
1 Parent(s): d1c0af9

send test file

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. test.py +20 -0
README.md CHANGED
@@ -7,7 +7,7 @@ colorTo: green
7
  sdk: gradio
8
  sdk_version: 3.47.1
9
  python_version: 3.8
10
- app_file: app.py
11
  license: other
12
  pinned: false
13
  ---
 
7
  sdk: gradio
8
  sdk_version: 3.47.1
9
  python_version: 3.8
10
+ app_file: test.py
11
  license: other
12
  pinned: false
13
  ---
test.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ app to test the environment of hugging face
3
+ """
4
+ import os, sys
5
+ import gradio as gr
6
+ import torch
7
+
8
+
9
+ if __name__ == '__main__':
10
+ # debug print
11
+ print("CUDA_HOME:", os.getenv('CUDA_HOME'))
12
+ print("CUDA_PATH:", os.getenv("CUDA_PATH"))
13
+ print("CUDA available?", torch.cuda.is_available())
14
+ demo = gr.Blocks(title="HDM Interaction Reconstruction Demo")
15
+
16
+ with demo:
17
+ input_rgb = gr.Image(label='Input RGB', type='numpy')
18
+
19
+
20
+ demo.queue().launch()