Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -201,8 +201,8 @@ def process(query_img, state,
         wrap_fp16_model(model)
     load_checkpoint(model, checkpoint_path, map_location='cpu')
 	
-	data["img_s"] = data["img_s"][0].cpu().numpy().tolist()
-	data["img_q"] = data["img_q"].cpu().numpy().tolist()
+    data["img_s"] = data["img_s"][0].cpu().numpy().tolist()
+    data["img_q"] = data["img_q"].cpu().numpy().tolist()
     data['target_weight_s'][0] = data['target_weight_s'][0].cpu().numpy().tolist()
     data['target_s'][0] = data['target_s'][0].cpu().numpy().tolist()
     
@@ -212,7 +212,7 @@ def process(query_img, state,
     data['img_metas'][0]['query_center'] = data['img_metas'][0]['query_center'].cpu().tolist()
     data['img_metas'][0]['sample_scale'][0] = data['img_metas'][0]['sample_scale'][0].cpu().tolist()
     data['img_metas'][0]['query_scale'] = data['img_metas'][0]['query_scale'].cpu().tolist()
-	
+    
     model.eval()
     
     str_data = json.dumps(data, cls=CustomEncoder)