""" 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()