jiayong commited on
Commit
8a10867
1 Parent(s): ca9a0da

Update gen_client.py

Browse files

update get_ranking_location

Files changed (1) hide show
  1. gen_client.py +7 -2
gen_client.py CHANGED
@@ -642,14 +642,19 @@ class HumanGenService:
642
  startt = self.all_requests_time[req]
643
  waste_time = (endt - startt)/60
644
  if waste_time > avg_process_time + 1:
 
 
 
 
 
645
  data = '{"header":{"request_id":"","service_id":"","task_id":""},"payload":{"input": {"ref_image_path": "", "ref_video_path": "", "ref_video_name": "", "input_prompt": "", "prompt_template": "", "scale_depth": 0.7, "scale_pose": 0.5},"parameters":{}}}'
646
  data = json.loads(data) # string to dict
647
  data['header']['service_id'] = DASHONE_SERVICE_ID
648
  data['header']['request_id'] = req
649
  data['header']['task_id'] = req
650
  data['header']['attributes'] = {}
651
- data['header']['attributes']['user_id'] = user_id
652
- data['payload']['input']['user_id'] = user_id
653
  data = json.dumps(data) # to string
654
  ret_status, ret_json = query_video_generation(request_id=req, data=data)
655
  # print(f'ret_json = {ret_json}')
 
642
  startt = self.all_requests_time[req]
643
  waste_time = (endt - startt)/60
644
  if waste_time > avg_process_time + 1:
645
+ uuid = ''
646
+ for uid in self.all_user_requests:
647
+ if req in self.all_user_requests[uid]:
648
+ uuid = uid
649
+ break
650
  data = '{"header":{"request_id":"","service_id":"","task_id":""},"payload":{"input": {"ref_image_path": "", "ref_video_path": "", "ref_video_name": "", "input_prompt": "", "prompt_template": "", "scale_depth": 0.7, "scale_pose": 0.5},"parameters":{}}}'
651
  data = json.loads(data) # string to dict
652
  data['header']['service_id'] = DASHONE_SERVICE_ID
653
  data['header']['request_id'] = req
654
  data['header']['task_id'] = req
655
  data['header']['attributes'] = {}
656
+ data['header']['attributes']['user_id'] = uuid
657
+ data['payload']['input']['user_id'] = uuid
658
  data = json.dumps(data) # to string
659
  ret_status, ret_json = query_video_generation(request_id=req, data=data)
660
  # print(f'ret_json = {ret_json}')