Spaces:

fffiloni
/

KDTalker

Running on Zero

fffiloni commited on 18 days ago

Commit

9aee76a

verified ·

1 Parent(s): f2bc26c

Update gradio_app.py

Files changed (1) hide show

gradio_app.py CHANGED Viewed

@@ -141,7 +141,7 @@ class Inferencer(object):
         from model import get_model
         self.point_diffusion = get_model()
-        ckpt = torch.load('./KDTalker.pth')
         self.point_diffusion.load_state_dict(ckpt['model'])
         self.point_diffusion.eval()

         from model import get_model
         self.point_diffusion = get_model()
+        ckpt = torch.load('ckpts/KDTalker.pth')
         self.point_diffusion.load_state_dict(ckpt['model'])
         self.point_diffusion.eval()