From c5ea0f00ffc5d599c133712b7e9a539d12d18a6d Mon Sep 17 00:00:00 2001
From: Jhin <49975880+649459021@users.noreply.github.com>
Date: Mon, 13 Sep 2021 19:56:01 +0800
Subject: [PATCH] Update demo_transfer.py

---
 demos/demo_transfer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/demos/demo_transfer.py b/demos/demo_transfer.py
index eea41d99..16c2ac27 100755
--- a/demos/demo_transfer.py
+++ b/demos/demo_transfer.py
@@ -13,6 +13,7 @@
 # For comments or questions, please email us at deca@tue.mpg.de
 # For commercial licensing contact, please contact ps-license@tuebingen.mpg.de
 
+import torch
 import os, sys
 import cv2
 import numpy as np