Spaces:
Paused
Paused
giorgio-caparvi commited on
Commit ·
e0e48b5
1
Parent(s): f5bc015
fix attribute
Browse files- 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(
|
| 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 |
-
|
| 96 |
-
|
| 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
|