giorgio-caparvi commited on
Commit
e0e48b5
·
1 Parent(s): f5bc015

fix attribute

Browse files
Files changed (1) hide show
  1. api/model/src/eval.py +3 -3
api/model/src/eval.py CHANGED
@@ -30,7 +30,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "true"
30
  os.environ["WANDB_START_METHOD"] = "thread"
31
 
32
 
33
- def main(im_sketch: io.BytesIO, json_data_from_req: dict) -> None:
34
  args = eval_parse_args()
35
  accelerator = Accelerator(
36
  mixed_precision=args.mixed_precision,
@@ -92,8 +92,8 @@ def main(im_sketch: io.BytesIO, json_data_from_req: dict) -> None:
92
  radius=5,
93
  tokenizer=tokenizer,
94
  size=(512, 384),
95
- im_sketch=im_sketch,
96
- json_data_from_req=json_data_from_req
97
  )
98
  else:
99
  raise NotImplementedError
 
30
  os.environ["WANDB_START_METHOD"] = "thread"
31
 
32
 
33
+ def main(im_sketch_buffer_from_request: io.BytesIO, json_from_req: dict) -> None:
34
  args = eval_parse_args()
35
  accelerator = Accelerator(
36
  mixed_precision=args.mixed_precision,
 
92
  radius=5,
93
  tokenizer=tokenizer,
94
  size=(512, 384),
95
+ im_sketch_buffer_from_request=im_sketch_buffer_from_request,
96
+ json_from_req=json_from_req
97
  )
98
  else:
99
  raise NotImplementedError