test.py

"""
app to test the environment of hugging face
"""
import os, sys
import gradio as gr
import torch


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