xiexh20's picture
add share control
a26c57b
"""
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()