Spaces:
Sleeping
Sleeping
File size: 487 Bytes
1963f53 a26c57b 1963f53 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
"""
app to test the environment of hugging face
"""
import os, sys
import gradio as gr
import torch
import pytorch3d
if __name__ == '__main__':
# debug print
print("CUDA_HOME:", os.getenv('CUDA_HOME'))
print("CUDA_PATH:", os.getenv("CUDA_PATH"))
print("CUDA available?", torch.cuda.is_available())
demo = gr.Blocks(title="HDM Interaction Reconstruction Demo")
with demo:
input_rgb = gr.Image(label='Input RGB', type='numpy')
demo.queue().launch() |