lime-j commited on
Commit
b58298b
·
1 Parent(s): fa1301c

small fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
102
  input_image = input_image.resize((size, int(size * input_image.height / input_image.width)))
103
  else:
104
  input_image = input_image.resize((int(size * input_image.width / input_image.height), size))
105
- tensor = TF.to_tensor(img)
106
 
107
  original_size = tensor.shape[-2:]
108
  padded_tensor, padding = pad_for_model(tensor, args.round_multiple)
 
102
  input_image = input_image.resize((size, int(size * input_image.height / input_image.width)))
103
  else:
104
  input_image = input_image.resize((int(size * input_image.width / input_image.height), size))
105
+ tensor = TF.to_tensor(input_image)
106
 
107
  original_size = tensor.shape[-2:]
108
  padded_tensor, padding = pad_for_model(tensor, args.round_multiple)