Update demo.py

demo.py

@@ -5,6 +5,7 @@ import csv
 import sys
 import warnings
 import random
+from pathlib import Path
 import gradio as gr
 import torch
 import torch.nn as nn
@@ -17,12 +18,15 @@ import time
 import pprint
 from loguru import logger
 import smplx
-
 import matplotlib.pyplot as plt
 from utils import config, logger_tools, other_tools_hf, metric, data_transfer, other_tools
 from utils.joints import upper_body_mask, hands_body_mask, lower_body_mask
 from dataloaders import data_tools
 from dataloaders.build_vocab import Vocab
+PROJECT_ROOT = Path(__file__).resolve().parent
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from optimizers.optim_factory import create_optimizer
 from optimizers.scheduler_factory import create_scheduler
 from optimizers.loss_factory import get_loss_func