tidalove commited on
Commit
667d357
·
verified ·
1 Parent(s): decbd98

changing indents bruh

Browse files
Files changed (1) hide show
  1. test_api.py +2 -2
test_api.py CHANGED
@@ -58,10 +58,10 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, vgg_pth='vgg_n
58
  # Timer
59
  times = []
60
 
61
- style_ds = load_dataset(style_dataset_pth, split="train")
62
  # do i need to stick a dataloader around this? idk
63
 
64
- for style_idx, style_item in enumerate(style_ds):
65
  style_img = style_item['image']
66
  print(style_img)
67
  style_tensor = t(style_img).unsqueeze(0).to(device)
 
58
  # Timer
59
  times = []
60
 
61
+ style_ds = load_dataset(style_dataset_pth, split="train")
62
  # do i need to stick a dataloader around this? idk
63
 
64
+ for style_idx, style_item in enumerate(style_ds):
65
  style_img = style_item['image']
66
  print(style_img)
67
  style_tensor = t(style_img).unsqueeze(0).to(device)