Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -112,7 +112,7 @@ def plot_results(support_img, query_img, support_kp, support_w, query_kp, query_
 def estimate(model, data):
     model.cuda()
     with torch.no_grad():
-        return model(data)
+        return model(**data)
 
 def process(query_img, state,
             cfg_path='configs/test/1shot_split1.py',