xiexh20's picture
add share control
a26c57b
raw
history blame
487 Bytes
"""
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()