Upload app_hg.py with huggingface_hub

app_hg.py

@@ -100,6 +100,10 @@ print(glob("/usr/local/cuda/*"))
 print(torch.cuda.is_available())
 print(os.environ.get('CUDA_HOME', None))
 os.environ['CUDA_HOME'] = '/usr/local/cuda'
+# Optionally, update PATH and LD_LIBRARY_PATH if needed
+os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
+os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
+
 ################################################################
 # initial setting
 ################################################################