Spaces:

capagio
/

garment_designer

Paused

giorgio-caparvi commited on Sep 1, 2024

Commit

e9c985d

1 Parent(s): 9aabd56

passing caption through json

Files changed (1) hide show

api/model/src/eval.py CHANGED Viewed

@@ -87,7 +87,8 @@ def main(json_from_req: dict) -> None:
             sketch_threshold_range=(20, 20),
             tokenizer=tokenizer,
             category=category,
-            size=(512, 384)
         )
     elif args.dataset == "vitonhd":
         test_dataset = VitonHDDataset(

             sketch_threshold_range=(20, 20),
             tokenizer=tokenizer,
             category=category,
+            size=(512, 384),
+            json_from_req=json_from_req
         )
     elif args.dataset == "vitonhd":
         test_dataset = VitonHDDataset(